| import subprocess, sys, os |
| import tyro |
| from pixel3dmm import env_paths |
|
|
| def run_and_check(cmd, cwd=None): |
| print(f"> {' '.join(cmd)} (in {cwd or os.getcwd()})") |
| |
| result = subprocess.run( |
| cmd, |
| cwd=cwd, |
| check=True, |
| ) |
| return result |
|
|
| def main(video_or_images_path: str): |
| vid_name = ( |
| os.path.basename(video_or_images_path) |
| if os.path.isdir(video_or_images_path) |
| else os.path.splitext(os.path.basename(video_or_images_path))[0] |
| ) |
|
|
| SCRIPTS = os.path.join(env_paths.CODE_BASE, "scripts") |
| MICA = os.path.join(env_paths.CODE_BASE, "src", "pixel3dmm", "preprocessing", "MICA") |
|
|
| run_and_check( |
| [sys.executable, "-u", "run_cropping.py", "--video_or_images_path", video_or_images_path], |
| cwd=SCRIPTS, |
| ) |
| run_and_check( |
| [sys.executable, "-u", "demo.py", "-video_name", vid_name], |
| cwd=MICA, |
| ) |
| if __name__ == "__main__": |
| tyro.cli(main) |
|
|
