Skip to content

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

c

parent 590cd6dd
Pipeline #349 canceled with stages
      Show whitespace changes
      Inline Side-by-side
      Showing with 3 additions and 3 deletions
      testVideo.py
      View file @ d776a1e5
      ...@@ -31,7 +31,7 @@ def post_precess(img, wh): ...@@ -31,7 +31,7 @@ def post_precess(img, wh):
      def save(frame, path, name): def save(frame, path, name):
      img_name = name + '.png' img_name = name + '.png'
      img = Image.fromarray(frame.astype('uint8')) # convert image to uint8: frame.astype('uint8') img = Image.fromarray(frame.astype('uint8')) # convert image to uint8: frame.astype('uint8')
      img.save(path + img_name) img.save(path + '/' + img_name)
      def load_image(image_path, x32=False): def load_image(image_path, x32=False):
      ...@@ -155,9 +155,9 @@ def video2comix(args): ...@@ -155,9 +155,9 @@ def video2comix(args):
      style_frame.save(os.path.join(args.output_dir, imgname + str(i) + "_c.jpg")) 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.astype('uint8') #img = style_frame.astype('uint8')
      flow_video.write(img) flow_video.write(style_frame)
      # flow_video.write(cv2.cvtColor(img,cv2.COLOR_BGR2RGB)) # flow_video.write(cv2.cvtColor(img,cv2.COLOR_BGR2RGB))
      # cv2.waitKey(1) # cv2.waitKey(1)
      i = i + 1 i = i + 1
      ......
      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