#!/bin/bash
# BEVFusion 推理结果可视化脚本

cd /workspace/bevfusion

export PATH=/opt/conda/bin:$PATH

echo "======================================"
echo "   BEVFusion 可视化工具"
echo "======================================"
echo ""
echo "请选择可视化模式："
echo "1) 快速可视化 (10个样本图像)"
echo "2) 生成完整视频"
echo "3) 全部生成 (图像 + 视频)"
echo ""
read -p "请输入选项 [1-3]: " choice

case $choice in
    1)
        echo "开始快速可视化..."
        python visualize_results_with_video.py \
            --result-file results_epoch19.pkl \
            --nusc-root data/nuscenes \
            --mode quick \
            --num-samples 10 \
            --output-dir visualizations
        ;;
    2)
        echo "开始生成视频..."
        python visualize_results_with_video.py \
            --result-file results_epoch19.pkl \
            --nusc-root data/nuscenes \
            --mode video \
            --video-fps 10 \
            --output-dir visualizations
        ;;
    3)
        echo "生成图像和视频..."
        python visualize_results_with_video.py \
            --result-file results_epoch19.pkl \
            --nusc-root data/nuscenes \
            --mode all \
            --num-samples 20 \
            --video-fps 10 \
            --output-dir visualizations
        ;;
    *)
        echo "无效选项！"
        exit 1
        ;;
esac

echo ""
echo "可视化完成！结果保存在: visualizations/"
echo "======================================"