File size: 1,060 Bytes
a7aea10 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
#!/bin/bash
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"
|