Fix path of mediapipe FLAME embedding

capjason · Feb 13, 2023 · 9e14fa6 · 9e14fa6
1 parent b3dfa96
commit 9e14fa6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/gdl_apps/EMOCA/utils/load.py b/gdl_apps/EMOCA/utils/load.py
@@ -145,6 +145,7 @@ def replace_asset_dirs(cfg, output_dir : Path, ):
                 "FLAME/geometry/fixed_uv_displacements/fixed_displacement_256.npy")
         cfg[mode].model.flame_model_path = str(asset_dir / "FLAME/geometry/generic_model.pkl")
         cfg[mode].model.flame_lmk_embedding_path = str(asset_dir / "FLAME/geometry/landmark_embedding.npy")
+        cfg[mode].model.flame_mediapipe_lmk_embedding_path = str(asset_dir / "FLAME/geometry/mediapipe_landmark_embedding.npz")
         cfg[mode].model.face_mask_path = str(asset_dir / "FLAME/mask/uv_face_mask.png")
         cfg[mode].model.face_eye_mask_path  = str(asset_dir / "FLAME/mask/uv_face_eye_mask.png")
         cfg[mode].model.pretrained_modelpath = str(asset_dir / "DECA/data/deca_model.tar")