alexnasa commited on
Commit
2c99211
·
verified ·
1 Parent(s): 0634bc4

Update src/pixel3dmm/preprocessing/MICA/demo.py

Browse files
src/pixel3dmm/preprocessing/MICA/demo.py CHANGED
@@ -22,6 +22,7 @@ import traceback
22
  from glob import glob
23
  from pathlib import Path
24
  from PIL import Image
 
25
 
26
  import cv2
27
  import numpy as np
@@ -173,7 +174,7 @@ if __name__ == '__main__':
173
  parser = argparse.ArgumentParser(description='MICA - Towards Metrical Reconstruction of Human Faces')
174
  parser.add_argument('-video_name', required=True, type=str)
175
  parser.add_argument('-a', default='demo/arcface', type=str, help='Processed images for MICA input')
176
- parser.add_argument('-m', default='data/pretrained/mica.tar', type=str, help='Pretrained model path')
177
 
178
  args = parser.parse_args()
179
  cfg = get_cfg_defaults()
 
22
  from glob import glob
23
  from pathlib import Path
24
  from PIL import Image
25
+ from pixel3dmm import env_paths
26
 
27
  import cv2
28
  import numpy as np
 
174
  parser = argparse.ArgumentParser(description='MICA - Towards Metrical Reconstruction of Human Faces')
175
  parser.add_argument('-video_name', required=True, type=str)
176
  parser.add_argument('-a', default='demo/arcface', type=str, help='Processed images for MICA input')
177
+ parser.add_argument('-m', default=f'{env_paths.MICA_TAR_ASSET}', type=str, help='Pretrained model path')
178
 
179
  args = parser.parse_args()
180
  cfg = get_cfg_defaults()