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