alexnasa commited on
Commit
1dbf56f
·
1 Parent(s): 5cf3b22

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.FLAME_ASSETS}', 'rb') as f:
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.FLAME_ASSETS}/FLAME2020/landmark_embedding.npy', allow_pickle=True, encoding='latin1')
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