Skip to content

A
animegan2-pytorch
Commit 0513d0f1 authored by mac's avatar mac
Browse files
Options

c

parent 6f66dc41
Pipeline #367 canceled with stages
      Hide whitespace changes
      Inline Side-by-side
      Showing with 2 additions and 2 deletions
      testVideo.py
      View file @ 0513d0f1
      ......@@ -194,12 +194,12 @@ def video2comix(args):
      #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')
      img = style_frame #.astype('uint8')
      #print('---------style_frame type=' , type(style_frame))
      print('----------one frame time:',stylization_time,', style_frame type:',type(style_frame))
      #img = np.asarray(style_frame)
      #flow_video.write(img)
      flow_video.write(cv2.cvtColor(img,cv2.COLOR_RGB2BGR)) #COLOR_BGR2RGB
      flow_video.write(cv2.cvtColor(img,cv2.COLOR_RGB2BGR).astype('uint8')) #COLOR_BGR2RGB
      # cv2.waitKey(1)
      i = i + 1
      if i > 3:
      ......
      Markdown is supported
      0% or
      You are about to add 0 people to the discussion. Proceed with caution.
      Finish editing this message first!
      Please register or to comment