Merge branch 'wyhsirius:main' into replicate

wyhsirius · Jun 29, 2022 · dad4a73 · dad4a73
2 parents 114afee + 39bea53
commit dad4a73
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -9,6 +9,7 @@ This is the official PyTorch implementation of the ICLR 2022 paper "Latent Image
 - PyTorch 1.5+
 - tensorboard
 - moviepy
+- av
 - tqdm
 - lpips
 

diff --git a/res/results will be saved here b/res/results will be saved here
@@ -0,0 +1 @@
+
diff --git a/run_demo.py b/run_demo.py
@@ -66,8 +66,9 @@ def __init__(self, args):
         self.gen.eval()
 
         print('==> loading data')
-        self.save_path = os.path.join(args.save_folder + '/%s' % args.model,
-                                      Path(args.source_path).stem + '_' + Path(args.driving_path).stem + '.mp4')
+        self.save_path = args.save_folder + '/%s' % args.model
+        os.makedirs(self.save_path, exist_ok=True)
+        self.save_path = os.path.join(self.save_path, Path(args.source_path).stem + '_' + Path(args.driving_path).stem + '.mp4')
         self.img_source = img_preprocessing(args.source_path, args.size).cuda()
         self.vid_target, self.fps = vid_preprocessing(args.driving_path)
         self.vid_target = self.vid_target.cuda()
-Original file line number
+Diff line change
@@ Expand Up @@
     - PyTorch 1.5+
     - tensorboard
     - moviepy
+    - av
     - tqdm
     - lpips
@@ Expand Down @@