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