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