Skip to content

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

c

parent c1cd06e9
Pipeline #346 canceled with stages
      Hide whitespace changes
      Inline Side-by-side
      Showing with 4 additions and 3 deletions
      testVideo.py
      View file @ 14a7cd75
      ...@@ -49,8 +49,8 @@ def load_image(image_path, x32=False): ...@@ -49,8 +49,8 @@ def load_image(image_path, x32=False):
      def to_32s(x): def to_32s(x):
      return 256 if x < 256 else x - x % 32 return 256 if x < 256 else x - x % 32
      def x32(img, x32=False): def img_2_x32(img, x32=False):
      img = img.convert("RGB") #img = img.convert("RGB")
      if x32: if x32:
      # def to_32s(x): # def to_32s(x):
      ...@@ -72,7 +72,8 @@ def img2Comix(args,net,device,image): ...@@ -72,7 +72,8 @@ def img2Comix(args,net,device,image):
      #image = load_image(os.path.join(args.input_dir, image_name), args.x32) #image = load_image(os.path.join(args.input_dir, image_name), args.x32)
      image = x32(image,args.x32) image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
      image = img_2_x32(image,args.x32)
      with torch.no_grad(): with torch.no_grad():
      image = to_tensor(image).unsqueeze(0) * 2 - 1 image = to_tensor(image).unsqueeze(0) * 2 - 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