Spaces:

alexnasa
/

pixel3dmm

Running on Zero

alexnasa commited on 10 days ago

Commit

51841fe

verified ·

1 Parent(s): 522bbb9

Update src/pixel3dmm/tracking/tracker.py

Files changed (1) hide show

src/pixel3dmm/tracking/tracker.py CHANGED Viewed

@@ -205,7 +205,7 @@ class Tracker(object):
         self.R_base = {}
         self.t_base = {}
-        flame_mesh_mask = np.load(f'{env_paths.FLAME_ASSET}/FLAME2020/FLAME_masks/FLAME_masks.pkl', allow_pickle=True, encoding='latin1')
         self.vertex_face_mask = torch.from_numpy(flame_mesh_mask['face']).cuda().long()

         self.R_base = {}
         self.t_base = {}
+        flame_mesh_mask = np.load(f'{env_paths.FLAME_MASK_ASSET}/FLAME2020/FLAME_masks/FLAME_masks.pkl', allow_pickle=True, encoding='latin1')
         self.vertex_face_mask = torch.from_numpy(flame_mesh_mask['face']).cuda().long()