#!/bin/bash
# 快速测试推理脚本（只推理1个样本）

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

echo "=================================================="
echo "快速测试推理脚本"
echo "推理1个样本验证功能"
echo "=================================================="
echo ""

# 检查checkpoint
if [ ! -f "runs/run-326653dc-74184412/epoch_19.pth" ]; then
    echo "❌ epoch_19.pth不存在"
    exit 1
fi

echo "✅ Checkpoint文件存在"
echo ""

# 快速测试（只推理1个样本）
/opt/conda/bin/python inference_and_visualize.py \
    --samples 1 \
    --output-dir inference_test \
    --show-score-thr 0.3

echo ""
echo "=================================================="
echo "测试完成！检查输出："
echo "  ls -lh inference_test/"
echo "=================================================="