c

testVideo.py

@@ -99,10 +99,14 @@ def img2Comix(args,net,device,image):
 
 
     #image = load_image(os.path.join(args.input_dir, image_name), args.x32)
+    w, h = image.size
+    print('>>input img size=', w, ',', h)
     image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
-    print('>>input img size=',image.size.w,',',image.size.h)
+    w, h = image.size
+    print('>>cvtColor img size=',w,',',h)
     image = img_2_x32(image,args.x32)
-    print('>>img_2_x32 size=',image.size.w,',',image.size.h)
+    w, h = image.size
+    print('>>img_2_x32 size=', w, ',', h)
     with torch.no_grad():
         #print('----------torch.no_grad() handle')
         #变成tensor数据后，形状变为CxHxW
@@ -112,14 +116,14 @@ def img2Comix(args,net,device,image):
         # out = to_pil_image(out)
 
         out = (out).numpy()
-        print(out.shape)
+        #print(out.shape)
         # 改变矩阵形状，因为cv2的图片格式是HxWxC，使用转置transpose
         # switch channels -> (c, h, w) -> (h, w, c)
         out = np.rollaxis(out, 0, 3) #形状变为HxWxC
         # 改变数值，矩阵乘以255
         out = 255 * out
-        print(type(out))
-        print(out.shape)
+        #print(type(out))
+        #print(out.shape)
         #print(out)
 
     return out