76 lines
1.9 KiB
Batchfile
76 lines
1.9 KiB
Batchfile
@echo off
|
|
chcp 65001 >nul
|
|
cls
|
|
|
|
echo ============================================================
|
|
echo 导出 ONNX 模型 (YOLOv8)
|
|
echo ============================================================
|
|
echo.
|
|
|
|
:: 设置模型路径
|
|
set MODEL_PATH=runs/detect/train/weights/best.pt
|
|
|
|
:: 检查模型是否存在
|
|
if not exist %MODEL_PATH% (
|
|
echo [错误] 找不到模型文件: %MODEL_PATH%
|
|
echo.
|
|
echo 请先训练模型:
|
|
echo 运行 02_train.bat
|
|
pause
|
|
exit /b 1
|
|
)
|
|
|
|
echo [信息] 输入模型: %MODEL_PATH%
|
|
echo.
|
|
|
|
:: 导出 ONNX
|
|
echo ============================================================
|
|
echo 导出 ONNX
|
|
echo ============================================================
|
|
echo.
|
|
|
|
yolo export model=%MODEL_PATH% format=onnx imgsz=640 opset=12 simplify=True
|
|
|
|
if %ERRORLEVEL% neq 0 (
|
|
echo.
|
|
echo [错误] 导出失败！
|
|
pause
|
|
exit /b 1
|
|
)
|
|
|
|
echo.
|
|
echo ============================================================
|
|
echo 导出完成！
|
|
echo ============================================================
|
|
echo.
|
|
|
|
:: 检查输出文件
|
|
set ONNX_PATH=runs\detect\train\weights\best.onnx
|
|
if exist %ONNX_PATH% (
|
|
echo [成功] ONNX 模型: %ONNX_PATH%
|
|
|
|
:: 获取文件大小
|
|
for %%I in (%ONNX_PATH%) do (
|
|
set SIZE=%%~zI
|
|
)
|
|
echo [信息] 文件大小: %SIZE% bytes
|
|
) else (
|
|
echo [警告] 找不到输出文件
|
|
)
|
|
|
|
echo.
|
|
echo ============================================================
|
|
echo 下一步操作
|
|
echo ============================================================
|
|
echo.
|
|
echo 1. 复制 ONNX 文件到 Ubuntu 机器:
|
|
echo scp %ONNX_PATH% user@ubuntu-pc:~/rknn_convert/
|
|
echo.
|
|
echo 2. 在 Ubuntu 上转换为 RKNN:
|
|
echo python 04_convert_rknn.py best.onnx -o shoe_detector.rknn -t rk3588
|
|
echo.
|
|
echo 3. 部署到 RK3588:
|
|
echo scp shoe_detector.rknn orangepi@^<rk3588_ip^>:/home/orangepi/apps/OrangePi3588Media/models/
|
|
echo.
|
|
pause
|