#!/bin/bash
# 简化版推理脚本 - 使用mmdet3d标准测试工具

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

echo "=================================================="
echo "BEVFusion推理（使用mmdet3d标准工具）"
echo "=================================================="
echo "Checkpoint: epoch_19.pth"
echo "配置: multitask.yaml"
echo "数据集: nuScenes val (前10个样本)"
echo "=================================================="
echo ""

CONFIG="configs/nuscenes/det/transfusion/secfpn/camera+lidar/swint_v0p075/multitask.yaml"
CHECKPOINT="runs/run-326653dc-74184412/epoch_19.pth"

# 检查文件
if [ ! -f "$CHECKPOINT" ]; then
    echo "❌ Checkpoint不存在: $CHECKPOINT"
    exit 1
fi

echo "✅ 文件检查通过"
echo ""
echo "开始推理..."
echo ""

# 使用mmdet3d的test.py进行推理和评估
/opt/conda/bin/torchpack dist-run -np 1 /opt/conda/bin/python tools/test.py \
    "$CONFIG" \
    "$CHECKPOINT" \
    --eval bbox segm \
    --show-dir inference_results_simple

echo ""
echo "=================================================="
echo "✅ 推理完成！"
echo "结果保存在: inference_results_simple/"
echo "评估指标已输出"
echo "=================================================="