c

testVideo.py

@@ -100,9 +100,9 @@ def img2Comix(args,net,device,image):
 
     #image = load_image(os.path.join(args.input_dir, image_name), args.x32)
     image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
-    print('>>input img size=',image.size)
+    print('>>input img size=',image.size.w,',',image.size.h)
     image = img_2_x32(image,args.x32)
-    print('>>img_2_x32 size=',image.size)
+    print('>>img_2_x32 size=',image.size.w,',',image.size.h)
     with torch.no_grad():
         #print('----------torch.no_grad() handle')
         #变成tensor数据后，形状变为CxHxW
@@ -189,7 +189,8 @@ def video2comix(args):
             break
         if su:
 
-            #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)