c

testVideo.py

@@ -109,6 +109,7 @@ def img2Comix(args,net,device,image):
         out = out.squeeze(0).clip(-1, 1) * 0.5 + 0.5
 
         out = (out).numpy()
+        print(out.shape)
         # 改变矩阵形状，因为cv2的图片格式是HxWxC，使用转置transpose
         # switch channels -> (c, h, w) -> (h, w, c)
         out = np.rollaxis(out, 0, 3) #形状变为HxWxC