单目3D初始代码

eval_tools/heading_analysis/test_visualize_heading.sh

@@ -0,0 +1,62 @@
+#!/bin/bash
+# Test script for visualizing heading errors
+
+
+# Set PYTHONPATH to project root for module imports
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+PROJECT_ROOT="$(cd "${SCRIPT_DIR}/../.." && pwd)"
+export PYTHONPATH="${PROJECT_ROOT}:${PYTHONPATH}"
+echo "========================================="
+
+echo "Heading Error Visualization Test"
+echo "========================================="
+echo ""
+
+INPUT_FILE="bad_heading_cases_top50.json"
+IMAGE_ROOT="/path/to/your/image/root"  # UPDATE THIS
+OUTPUT_DIR="runs/heading_viz_test"
+
+# Check if input file exists
+if [ ! -f "$INPUT_FILE" ]; then
+    echo "Error: Input file not found: $INPUT_FILE"
+    echo "Please run extract_bad_heading_cases.py first"
+    exit 1
+fi
+
+echo "Test 1: Generate all visualization types (first 10 cases)"
+echo "-----------------------------------------------------"
+python eval_tools/heading_analysis/visualize_heading_errors.py \
+    --input "$INPUT_FILE" \
+    --image-root "$IMAGE_ROOT" \
+    --output "$OUTPUT_DIR/all_viz" \
+    --max-cases 10 \
+    --viz-types bev image angle combined
+
+echo ""
+echo ""
+echo "Test 2: Generate only BEV views (first 20 cases)"
+echo "-----------------------------------------------------"
+python eval_tools/heading_analysis/visualize_heading_errors.py \
+    --input "$INPUT_FILE" \
+    --image-root "$IMAGE_ROOT" \
+    --output "$OUTPUT_DIR/bev_only" \
+    --max-cases 20 \
+    --viz-types bev
+
+echo ""
+echo ""
+echo "Test 3: Generate combined views (first 5 cases, high DPI)"
+echo "-----------------------------------------------------"
+python eval_tools/heading_analysis/visualize_heading_errors.py \
+    --input "$INPUT_FILE" \
+    --image-root "$IMAGE_ROOT" \
+    --output "$OUTPUT_DIR/combined_hq" \
+    --max-cases 5 \
+    --viz-types combined \
+    --dpi 150
+
+echo ""
+echo "========================================="
+echo "All tests complete!"
+echo "Output directory: $OUTPUT_DIR"
+echo "========================================="