Update src/pixel3dmm/utils/masking.py

src/pixel3dmm/utils/masking.py

@@ -48,7 +48,7 @@ class Struct(object):
 class Masking(nn.Module):
     def __init__(self):
 
-        dir = f'{env_paths.FLAME_ASSET}'
+        dir = f'{env_paths.FLAME_MASK_ASSET}'
         super(Masking, self).__init__()
         with open(f'{dir}/FLAME2020/FLAME_masks/FLAME_masks.pkl', 'rb') as f:
             ss = pickle.load(f, encoding='latin1')