Parse command-line arguments.
()
| 43 | framerate = 30 |
| 44 | |
| 45 | def parse_args(): |
| 46 | """Parse command-line arguments.""" |
| 47 | parser = argparse.ArgumentParser( |
| 48 | description="Run 3D point cloud inference and visualization using ARCroco3DStereo." |
| 49 | ) |
| 50 | parser.add_argument( |
| 51 | "--model_path", |
| 52 | type=str, |
| 53 | default="src/cut3r_512_dpt_4_64.pth", |
| 54 | help="Path to the pretrained model checkpoint.", |
| 55 | ) |
| 56 | parser.add_argument( |
| 57 | "--seq_path", |
| 58 | type=str, |
| 59 | default="", |
| 60 | help="Path to the directory containing the image sequence.", |
| 61 | ) |
| 62 | parser.add_argument( |
| 63 | "--device", |
| 64 | type=str, |
| 65 | default="cuda", |
| 66 | help="Device to run inference on (e.g., 'cuda' or 'cpu').", |
| 67 | ) |
| 68 | parser.add_argument( |
| 69 | "--size", |
| 70 | type=int, |
| 71 | default="512", |
| 72 | help="Shape that input images will be rescaled to; if using 224+linear model, choose 224 otherwise 512", |
| 73 | ) |
| 74 | parser.add_argument( |
| 75 | "--vis_threshold", |
| 76 | type=float, |
| 77 | default=1.5, |
| 78 | help="Visualization threshold for the point cloud viewer. Ranging from 1 to INF", |
| 79 | ) |
| 80 | parser.add_argument( |
| 81 | "--output_dir", |
| 82 | type=str, |
| 83 | default="./demo_tmp", |
| 84 | help="value for tempfile.tempdir", |
| 85 | ) |
| 86 | parser.add_argument( |
| 87 | "--port", |
| 88 | type=int, |
| 89 | default=7860, |
| 90 | help="port for the point cloud viewer", |
| 91 | ) |
| 92 | parser.add_argument( |
| 93 | "--model_update_type", |
| 94 | type=str, |
| 95 | default="cut3r", |
| 96 | help="model update type: cut3r or ttt3r", |
| 97 | ) |
| 98 | parser.add_argument( |
| 99 | "--frame_interval", |
| 100 | type=int, |
| 101 | default=1, |
| 102 | help="Frame interval for video processing (e.g., 1 means every frame, 2 means every other frame)", |