pixel3dmm / scripts /run_preprocessing.py
alexnasa's picture
Upload 66 files
cf92dec verified
raw
history blame
666 Bytes
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)