filename

testVideo.py

@@ -131,6 +131,10 @@ def img2Comix(args,net,device,image):
 
 @profile
 def video2comix(args):
+    if args.filename == '':
+        print('miss param: filename')
+        return
+
     device = args.device
 
     net = Generator()
@@ -142,9 +146,9 @@ def video2comix(args):
 
 
 
-    inputmp4 = jj(args.input_dir, 'f1_short.mp4')
-    outputmp4 = jj(args.output_dir, 'f1_short_comi.mp4')
-    outputh264 = jj(args.output_dir, 'f1_short_comi_h264.mp4')
+    inputmp4 = jj(args.input_dir, args.filename + '.mp4')
+    outputmp4 = jj(args.output_dir, args.filename + '.mp4')
+    outputh264 = jj(args.output_dir, args.filename + '_h264.mp4')
 
     vc = cv2.VideoCapture(inputmp4)
     print('video open status:', vc.isOpened())
@@ -235,6 +239,11 @@ if __name__ == '__main__':
         type=str, 
         default='./samples/inputs',
     )
+    parser.add_argument(
+        '--filename',
+        type=str,
+        default='',
+    )
     parser.add_argument(
         '--output_dir', 
         type=str,