Path problem fixed

OpenTalker · Jul 15, 2022 · 83b28c8 · 83b28c8
1 parent 0c9be1e
commit 83b28c8
Show file tree

Hide file tree

Showing 17 changed files with 19 additions and 714 deletions.
diff --git a/.idea/StyleHEAT.iml b/.idea/StyleHEAT.iml
diff --git a/.idea/misc.xml b/.idea/misc.xml
diff --git a/README.md b/README.md
@@ -40,8 +40,8 @@ Please download our [pre-trained model](https://drive.google.com/drive/folders/1
 |checkpoints/stylegan2_d_256.pth | Pre-trained StyleGAN discriminator.
 |checkpoints/model_ir_se50.pth | Pre-trained id-loss discriminator.
 |checkpoints/StyleHEAT_visual.pt | Pre-trained StyleHEAT model.
-|checkpoints/BFM.zip | 3DMM library. (Note the zip file should be unzipped to BFM/.)
-|checkpoints/epoch_20.pth | Pre-trained 3DMM extractor.
+|checkpoints/BFM | 3DMM library. (Note the zip file should be unzipped to BFM/.)
+|checkpoints/Deep3D/epoch_20.pth | Pre-trained 3DMM extractor.
 
 We also provide some example videos along with their corresponding 3dmm parameters in [videos.zip](https://drive.google.com/drive/folders/1-m47oPsa3kxjgK5eSJ8g8sHzG4zr2oRc?usp=sharing).
 Please unzip and put them in `docs/demo/videos/` for later inference.

diff --git a/configs/path.py b/configs/path.py
@@ -16,4 +16,7 @@
     'styleheat': 'checkpoints/StyleHEAT_visual.pt',
     # id_loss
     'irse50': 'checkpoints/model_ir_se50.pth',
+    # 3DMM
+    'BFM': 'checkpoints/BFM',
+    '3DMM': 'checkpoints/Deep3D/epoch_20.pth',
 }
diff --git a/loss/iterative_inversion_loss.py b/loss/iterative_inversion_loss.py
@@ -66,7 +66,6 @@ def cal_p_norm_loss(self, latent_in):
     def load_PCA_model(self):
         device = 'cuda'
         PCA_path = PRETRAINED_MODELS_PATH['FFHQ_PCA']
-        # PCA_path = '/apdcephfs/share_1290939/feiiyin/TH/Barbershop/pretrained_models/ffhq_PCA.npz'
 
         PCA_model = np.load(PCA_path)
         self.X_mean = torch.from_numpy(PCA_model['X_mean']).float().to(device)

diff --git a/loss/sync_loss.py b/loss/sync_loss.py
diff --git a/models/hfgi/test.py b/models/hfgi/test.py