no message

testVideo.py

@@ -99,14 +99,11 @@ def img2Comix(args,net,device,image):
 
 
     #image = load_image(os.path.join(args.input_dir, image_name), args.x32)
-    w, h, c = image.shape
-    print('>>input img size=', w, ',', h, ',',c)
+    print('>>input img size=', image.shape)
     image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
-    size = image.shape
-    print('>>cvtColor img size=',size)
+    #print('>>cvtColor img size=',image.shape)
     image = img_2_x32(image,args.x32)
-    size = image.shape
-    print('>>img_2_x32 size=', size)
+    print('>>img_2_x32 size=', image.shape)
     with torch.no_grad():
         #print('----------torch.no_grad() handle')
         #变成tensor数据后，形状变为CxHxW
@@ -203,8 +200,10 @@ def video2comix(args):
                 #save(style_frame, args.output_dir, imgname + str(i) + "_c")
 
             img = style_frame
+            #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))
+            print('----------one frame time:',stylization_time,', style_frame type:',type(style_frame),',size',img.shape)
             #img = np.asarray(style_frame)
             flow_video.write(img.astype('uint8'))
             #flow_video.write(cv2.cvtColor(img,cv2.COLOR_RGB2BGR).astype('uint8')) #COLOR_BGR2RGB