38 lines
835 B
Bash
38 lines
835 B
Bash
|
|
#!/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 "=================================================="
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|