39 lines
992 B
Batchfile
39 lines
992 B
Batchfile
@echo off
|
|
chcp 65001 >nul
|
|
cls
|
|
|
|
:: 设置 Python 3.11 路径
|
|
set "PATH=C:\Users\Tellme\AppData\Local\Programs\Python\Python311\Scripts;C:\Users\Tellme\AppData\Local\Programs\Python\Python311;%PATH%"
|
|
|
|
echo ============================================================
|
|
echo 导出 ONNX 模型 (640x640)
|
|
echo ============================================================
|
|
echo.
|
|
|
|
set MODEL_PATH=%USERPROFILE%\apps\ultralytics\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 模型: %USERPROFILE%\apps\ultralytics\runs\detect\train\weights\best.onnx
|
|
echo.
|
|
echo 下一步: 在 Ubuntu 上运行 04_convert_rknn.py 转换
|
|
echo.
|
|
pause
|