c

testVideo.py

@@ -76,6 +76,7 @@ def img2Comix(args,net,device,image):
     image = img_2_x32(image,args.x32)
 
     with torch.no_grad():
+        print('----------torch.no_grad() handle')
         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
@@ -145,12 +146,13 @@ def video2comix(args):
             print('video2comix - read frame is none')
             break
         if su:
+
+            save(frame, args.output_dir, imgname + str(i))
             # transform frame 2 comix
             # 风格化处理
             style_frame = img2Comix(args,net,device,frame)
-            save(frame, args.output_dir, imgname + str(i))
             if i == 0:
-                style_frame.save(os.path.join(args.output_dir, imgname + str(i) + "_c"))
+                style_frame.save(os.path.join(args.output_dir, imgname + str(i) + "_c.jpg"))
                 #save(style_frame, args.output_dir, imgname + str(i) + "_c")
 
             img = style_frame.astype('uint8')