单目3D初始代码

tools/scripts_for_gt/visualization/visualize_batch.sh

@@ -0,0 +1,54 @@
+#!/usr/bin/env bash
+# 对 DL_KPI_SCENE 下的所有 case 批量进行 GT 可视化
+# 用法:
+#   bash tools/scripts_for_gt/visualization/visualize_batch.sh [SAMPLE_INTERVAL]
+# 例:
+#   bash tools/scripts_for_gt/visualization/visualize_batch.sh 5   # 每隔5帧处理一帧
+
+set -euo pipefail
+
+SCENE_ROOT="/data1/dongying/Mono3d/G1Q3/dataset_for_evaluation/DL_KPI_SCENE"
+SAMPLE_INTERVAL="${1:-20}"   # 第一个位置参数为采样间隔，默认为20
+FILE_EXTENSION=".png"
+
+# 收集所有 case 目录（含 images/ 子目录的）
+mapfile -t CASE_DIRS < <(find "$SCENE_ROOT" -mindepth 1 -maxdepth 1 -type d | sort)
+
+echo "共找到 ${#CASE_DIRS[@]} 个 case，采样间隔=${SAMPLE_INTERVAL}"
+echo "======================================================"
+
+SUCCESS=0
+SKIP=0
+FAIL=0
+
+for CASE_DIR in "${CASE_DIRS[@]}"; do
+    IMAGE_DIR="${CASE_DIR}/images"
+    LABEL_DIR="${CASE_DIR}/labels_json"
+    OUTPUT_DIR="${CASE_DIR}/vis_json"
+
+    if [[ ! -d "$IMAGE_DIR" ]]; then
+        echo "[SKIP] ${CASE_DIR##*/} — 无 images/ 目录"
+        (( SKIP++ )) || true
+        continue
+    fi
+    if [[ ! -d "$LABEL_DIR" ]]; then
+        echo "[SKIP] ${CASE_DIR##*/} — 无 labels_json/ 目录"
+        (( SKIP++ )) || true
+        continue
+    fi
+
+    echo ""
+    echo ">>> 处理 case: ${CASE_DIR##*/}"
+    python tools/scripts_for_gt/visualization/visualize_batch.py \
+        --image-dir  "$IMAGE_DIR" \
+        --label-dir  "$LABEL_DIR" \
+        --output     "$OUTPUT_DIR" \
+        --file-extension "$FILE_EXTENSION" \
+        --sample-interval "$SAMPLE_INTERVAL" \
+    && (( SUCCESS++ )) || (( FAIL++ )) || true
+done
+
+echo ""
+echo "======================================================"
+echo "全部完成：成功=${SUCCESS}  跳过=${SKIP}  失败=${FAIL}" 
+