63 lines
1.9 KiB
Bash
Executable File
63 lines
1.9 KiB
Bash
Executable File
#!/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 "========================================="
|