55 lines
1.7 KiB
Bash
55 lines
1.7 KiB
Bash
|
#!/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}"
|
|||
|
|
|