C

testVideo.py

@@ -169,7 +169,7 @@ def video2comix(args):
 
 
     # MPEG can not play in web
-    fourcc = cv2.VideoWriter_fourcc('M','P','4','V') #H264 avc1
+    fourcc = cv2.VideoWriter_fourcc(*'MP4V') #H264 avc1
     wh = (frame_width, frame_height)
     #宽高32位格式化，不知道是不是会提高效率?
     if args.x32:
@@ -193,8 +193,8 @@ def video2comix(args):
             break
         if su:
 
-            if i==0:
-                save(frame, args.output_dir, imgname + str(i))
+            #if i==0:
+            #    save(frame, args.output_dir, imgname + str(i))
             # transform frame 2 comix
             # 风格化处理
             style_frame,stylization_time = img2Comix(args,net,device,frame)
@@ -206,14 +206,14 @@ def video2comix(args):
             #print('>>style_frame size=', img.shape)
 
             #print('---------style_frame type=' , type(style_frame))
-            print('----------one frame time:',stylization_time,', style_frame type:',type(style_frame),',size',style_frame.shape)
+            #print('----------one frame time:',stylization_time,', style_frame type:',type(style_frame),',size',style_frame.shape)
             #img = np.asarray(style_frame)
-            print(flow_video.write(style_frame.astype('uint8')))
+            flow_video.write(style_frame.astype('uint8'))
             #flow_video.write(cv2.cvtColor(img,cv2.COLOR_RGB2BGR).astype('uint8')) #COLOR_BGR2RGB
             # cv2.waitKey(1)
             i = i + 1
-            if i > 4:
-                break
+            #if i > 4:
+            #    break
     print('----------video saved: '+ outputmp4)
     flow_video.release()
     vc.release()