| IMAGE_DIR="your_input_path" | |
| OUTPUT_DIR="your_output_path" | |
| CUDA_DEVICE=0 | |
| # 1. run vggt.py | |
| CUDA_VISIBLE_DEVICES=$CUDA_DEVICE python3 vggt_infer.py --image_dir "$IMAGE_DIR" --output_dir "$OUTPUT_DIR/vggt" | |
| # # 2. run moge.py | |
| CUDA_VISIBLE_DEVICES=$CUDA_DEVICE python3 moge_infer.py --image_dir "$IMAGE_DIR" --output_dir "$OUTPUT_DIR/moge" | |
| # # 3. run metric3d.py | |
| INTRINSIC_PATH="$OUTPUT_DIR/vggt/colmap_data.json" | |
| CUDA_VISIBLE_DEVICES=$CUDA_DEVICE python3 metric3d_infer.py --image_dir "$IMAGE_DIR" --output_dir "$OUTPUT_DIR/metric3d" --intrinsic_path "$INTRINSIC_PATH" | |
| # # 4. conduct depth alignment | |
| MOGE_DEPTH_DIR="$OUTPUT_DIR/moge" | |
| VGGT_DEPTH_DIR="$OUTPUT_DIR/vggt" | |
| METRIC3D_DEPTH_DIR="$OUTPUT_DIR/metric3d" | |
| CUDA_VISIBLE_DEVICES=$CUDA_DEVICE python3 depth_align.py \ | |
| --image_dir "$IMAGE_DIR" \ | |
| --moge_depth_dir "$MOGE_DEPTH_DIR" \ | |
| --vggt_depth_dir "$VGGT_DEPTH_DIR/depth" \ | |
| --metric3d_depth_dir "$METRIC3D_DEPTH_DIR" \ | |
| --vggt_camera_json_file "$OUTPUT_DIR/vggt/colmap_data.json" \ | |
| --output_dir "$OUTPUT_DIR/final" | |