c

testVideo.py

@@ -99,14 +99,16 @@ 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)
+    #image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
     image = img_2_x32(image,args.x32)
 
     with torch.no_grad():
         #print('----------torch.no_grad() handle')
+        #变成tensor数据后，形状变为CxHxW
         image = to_tensor(image).unsqueeze(0) * 2 - 1
         out = net(image.to(device), args.upsample_align).cpu()
         out = out.squeeze(0).clip(-1, 1) * 0.5 + 0.5
+        # out = to_pil_image(out)
 
         out = (out).numpy()
         print(out.shape)
@@ -118,7 +120,7 @@ def img2Comix(args,net,device,image):
         print(type(out))
         print(out.shape)
         print(out)
-        #out = to_pil_image(out)
+
     return out