Spaces:
Running
on
Zero
Running
on
Zero
File size: 666 Bytes
cf92dec |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
import os
import tyro
from pixel3dmm import env_paths
def main(video_or_images_path : str):
if os.path.isdir(video_or_images_path):
vid_name = video_or_images_path.split('/')[-1]
else:
vid_name = video_or_images_path.split('/')[-1][:-4]
os.system(f'cd {env_paths.CODE_BASE}/scripts/ ; python run_cropping.py --video_or_images_path {video_or_images_path}')
os.system(f'cd {env_paths.CODE_BASE}/src/pixel3dmm/preprocessing/MICA ; python demo.py -video_name {vid_name}')
os.system(f'cd {env_paths.CODE_BASE}/scripts/ ; python run_facer_segmentation.py --video_name {vid_name}')
if __name__ == '__main__':
tyro.cli(main) |