From 3d1d9b037458d3f94f98f33321fc5e216beed4d0 Mon Sep 17 00:00:00 2001
From: in3omnia <coolcoolcoolboy@gmail.com>
Date: Wed, 10 Apr 2019 15:32:43 +0800
Subject: [PATCH] wrong args parse (#146)
in src/option.py line 151:
args.data_test = args.data_test.split('+')
so it should be
if args.data_test == ['video']:
---
src/main.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/main.py b/src/main.py
index fadbe65..5d3c3ea 100644
--- a/src/main.py
+++ b/src/main.py
@@ -12,7 +12,7 @@ checkpoint = utility.checkpoint(args)
def main():
global model
- if args.data_test == 'video':
+ if args.data_test == ['video']:
from videotester import VideoTester
model = model.Model(args, checkpoint)
t = VideoTester(args, model, checkpoint)
--
GitLab