c

testVideo.py

@@ -31,7 +31,7 @@ def post_precess(img, wh):
 def save(frame, path, name):
     img_name = name + '.png'
     img = Image.fromarray(frame.astype('uint8'))  # convert image to uint8:  frame.astype('uint8')
-    img.save(path + img_name)
+    img.save(path + '/' + img_name)
 
 
 def load_image(image_path, x32=False):
@@ -155,9 +155,9 @@ def video2comix(args):
                 style_frame.save(os.path.join(args.output_dir, imgname + str(i) + "_c.jpg"))
                 #save(style_frame, args.output_dir, imgname + str(i) + "_c")
 
-            img = style_frame.astype('uint8')
+            #img = style_frame.astype('uint8')
 
-            flow_video.write(img)
+            flow_video.write(style_frame)
             # flow_video.write(cv2.cvtColor(img,cv2.COLOR_BGR2RGB))
             # cv2.waitKey(1)
             i = i + 1