36 lines
762 B
Batchfile
36 lines
762 B
Batchfile
@echo off
|
|
chcp 65001 >nul
|
|
cls
|
|
|
|
echo ============================================================
|
|
echo 导出 ONNX 模型 (640x640)
|
|
echo ============================================================
|
|
echo.
|
|
|
|
set MODEL_PATH=runs/detect/train/weights/best.pt
|
|
|
|
if not exist %MODEL_PATH% (
|
|
echo [错误] 找不到模型: %MODEL_PATH%
|
|
echo 请先运行 02_train.bat 训练
|
|
pause
|
|
exit /b 1
|
|
)
|
|
|
|
echo [信息] 输入模型: %MODEL_PATH%
|
|
echo.
|
|
|
|
yolo export model=%MODEL_PATH% format=onnx imgsz=640 opset=12 simplify
|
|
|
|
if %ERRORLEVEL% neq 0 (
|
|
echo [错误] 导出失败！
|
|
pause
|
|
exit /b 1
|
|
)
|
|
|
|
echo.
|
|
echo [成功] ONNX 模型: runs/detect/train/weights/best.onnx
|
|
echo.
|
|
echo 下一步: 在 Ubuntu 上运行 04_convert_rknn.py 转换
|
|
echo.
|
|
pause
|