Spaces:
Running
on
Zero
Running
on
Zero
variable name fixed!
Browse files
src/pixel3dmm/preprocessing/MICA/utils/masking.py
CHANGED
@@ -49,7 +49,7 @@ class Masking(nn.Module):
|
|
49 |
ss = pickle.load(f, encoding='latin1')
|
50 |
self.masks = Struct(**ss)
|
51 |
|
52 |
-
with open(f'{env_paths.
|
53 |
ss = pickle.load(f, encoding='latin1')
|
54 |
flame_model = Struct(**ss)
|
55 |
|
|
|
49 |
ss = pickle.load(f, encoding='latin1')
|
50 |
self.masks = Struct(**ss)
|
51 |
|
52 |
+
with open(f'{env_paths.FLAME_ASSET}', 'rb') as f:
|
53 |
ss = pickle.load(f, encoding='latin1')
|
54 |
flame_model = Struct(**ss)
|
55 |
|
src/pixel3dmm/tracking/flame/FLAME.py
CHANGED
@@ -103,7 +103,7 @@ class FLAME(nn.Module):
|
|
103 |
self._register_default_params('expression_params', config.num_exp_params)
|
104 |
|
105 |
# Static and Dynamic Landmark embeddings for FLAME
|
106 |
-
lmk_embeddings = np.load(f'{env_paths.
|
107 |
lmk_embeddings = lmk_embeddings[()]
|
108 |
|
109 |
|
|
|
103 |
self._register_default_params('expression_params', config.num_exp_params)
|
104 |
|
105 |
# Static and Dynamic Landmark embeddings for FLAME
|
106 |
+
lmk_embeddings = np.load(f'{env_paths.FLAME_ASSET}/FLAME2020/landmark_embedding.npy', allow_pickle=True, encoding='latin1')
|
107 |
lmk_embeddings = lmk_embeddings[()]
|
108 |
|
109 |
|