c

testVideo.py

@@ -188,29 +188,29 @@ def video2comix(args):
             break
         if su:
 
-            save(frame, args.output_dir, imgname + str(i))
+            #save(frame, args.output_dir, imgname + str(i))
             # transform frame 2 comix
             # 风格化处理
             style_frame,stylization_time = img2Comix(args,net,device,frame)
-            if i == 0:
+            #if i == 0:
                 #style_frame.save(os.path.join(args.output_dir, imgname + str(i) + "_c.jpg"))
-                save(style_frame, args.output_dir, imgname + str(i) + "_c")
+                #save(style_frame, args.output_dir, imgname + str(i) + "_c")
 
-            img = style_frame #.astype('uint8')
+            img = style_frame
             #print('---------style_frame type=' , type(style_frame))
             print('----------one frame time:',stylization_time,', style_frame type:',type(style_frame))
             #img = np.asarray(style_frame)
-            #flow_video.write(img)
-            flow_video.write(cv2.cvtColor(img,cv2.COLOR_RGB2BGR).astype('uint8')) #COLOR_BGR2RGB
+            flow_video.write(img.astype('uint8'))
+            #flow_video.write(cv2.cvtColor(img,cv2.COLOR_RGB2BGR).astype('uint8')) #COLOR_BGR2RGB
             # cv2.waitKey(1)
             i = i + 1
-            if i > 3:
+            if i > 4:
                 break
     print('----------video saved: '+ outputmp4)
-    #call(["ffmpeg", "-i", outputmp4, "-vcodec", "libx264", "-f", "mp4", outputh264])
-    print('finish')
     flow_video.release()
     vc.release()
+    #call(["ffmpeg", "-i", outputmp4, "-vcodec", "libx264", "-f", "mp4", outputh264])
+    print('finish')
     return 0
 
 if __name__ == '__main__':