Spaces:

alexnasa
/

pixel3dmm

Running on Zero

alexnasa commited on 11 days ago

Commit

4569391

verified ·

1 Parent(s): e934072

Update src/pixel3dmm/tracking/flame/FLAME.py

Files changed (1) hide show

src/pixel3dmm/tracking/flame/FLAME.py CHANGED Viewed

@@ -103,7 +103,7 @@ class FLAME(nn.Module):
         self._register_default_params('expression_params', config.num_exp_params)
         # Static and Dynamic Landmark embeddings for FLAME
-        lmk_embeddings = np.load(f'{env_paths.FLAME_ASSET}/FLAME2020/landmark_embedding.npy', allow_pickle=True, encoding='latin1')
         lmk_embeddings = lmk_embeddings[()]

         self._register_default_params('expression_params', config.num_exp_params)
         # Static and Dynamic Landmark embeddings for FLAME
+        lmk_embeddings = np.load(f'{env_paths.FLAME_MASK_ASSET}/FLAME2020/landmark_embedding.npy', allow_pickle=True, encoding='latin1')
         lmk_embeddings = lmk_embeddings[()]