Spaces:
Running
on
Zero
Running
on
Zero
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='
|
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()
|