fix

scripts/anime.py

@@ -19,7 +19,7 @@ model = None
 def init_model(use_local=False):
     global model
     model_opt = "default"
-    device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+    device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') # issue: nevetherless, use_gpu is False, it still uses GPU
     model = create_model(model_opt, use_local).to(device)
     model.eval()
 

scripts/data.py

@@ -40,7 +40,7 @@ def get_transform(load_size=0, grayscale=False, method=bic, convert=True):
         transform_list.append(transforms.Grayscale(1))
     if load_size > 0:
         osize = [load_size, load_size]
-        transform_list.append(transforms.Resize(osize, method))
+        transform_list.append(transforms.Resize(osize, method, antialias=False))
     if convert:
         # transform_list += [transforms.ToTensor()]
         if grayscale:

scripts/model.py

@@ -154,8 +154,7 @@ def create_model(model, use_local):
 
     import os
     if model == 'default':
-        model_path = (lambda filename, subfolder: os.path.join(subfolder, filename) if use_local else download_file(filename, subfolder)) \
-                        ("netG.pth", "models/Anime2Sketch")
+        model_path = (lambda filename, subfolder: os.path.join(subfolder, filename) if use_local else download_file(filename, subfolder))("netG.pth", "models/Anime2Sketch")
         # model_path = ((filename, subfolder) => if (use_local) os.path.join(subfolder, filename) else download_file(filename, subfolder))("netG.pth", "models/Anime2Sketch") // JavaScript
 
         ckpt = torch.load(model_path)
@@ -176,8 +175,6 @@ def create_model(model, use_local):
             base = base.model[3]
 
         net.load_state_dict(ckpt)
-
-        os.chdir(cwd) # 元のディレクトリに戻る
     
     else:
         raise ValueError(f"model should be one of ['default', 'improved'], but got {model}")