c

testVideo.py

@@ -101,16 +101,16 @@ def img2Comix(args,net,device,image):
 
 
     #image = load_image(os.path.join(args.input_dir, image_name), args.x32)
-    print('>>input img size=', image.shape)
+    #print('>>input img size=', image.shape)
     image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
     #print('>>cvtColor img size=',image.shape)
     image = img_2_x32(image,args.x32)
-    print('>>img_2_x32 size=', image.shape)
+    #print('>>img_2_x32 size=', image.shape)
     with torch.no_grad():
         #print('----------torch.no_grad() handle')
         #变成tensor数据后，形状变为CxHxW
         image = to_tensor(image).unsqueeze(0) * 2 - 1
-        print('upsample_align=',args.upsample_align)
+        #print('upsample_align=',args.upsample_align)
         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)
@@ -129,7 +129,7 @@ def img2Comix(args,net,device,image):
     return out
 
 
-
+@profile
 def video2comix(args):
     device = args.device
 
@@ -198,15 +198,16 @@ def video2comix(args):
             # transform frame 2 comix
             # 风格化处理
             style_frame,stylization_time = img2Comix(args,net,device,frame)
-            if i == 0:
+            #if i == 0:
                 #style_frame.save(os.path.join(args.output_dir, imgname + str(i) + "_c.jpg"))
-                save(style_frame, args.output_dir, imgname + str(i) + "_c")
+                #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),',size',style_frame.shape)
+            if i == 0:
+                print('----------one frame time:',stylization_time,', style_frame type:',type(style_frame),',size',style_frame.shape)
             #img = np.asarray(style_frame)
             flow_video.write(style_frame.astype('uint8'))
             #flow_video.write(cv2.cvtColor(img,cv2.COLOR_RGB2BGR).astype('uint8')) #COLOR_BGR2RGB
@@ -264,4 +265,5 @@ if __name__ == '__main__':
 
     args = parser.parse_args()
     
-    video2comix(args)
+    time=video2comix(args)
+    print('totaltime:',time)