Spaces:
Paused
Paused
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()
|