Fix ROI training batch launcher
This commit is contained in:
tian
parent
36f6389d22
commit
6a3754421b
@ -1,92 +1,130 @@
|
|||||||
@echo off
|
@echo off
|
||||||
|
setlocal
|
||||||
chcp 65001 >nul
|
chcp 65001 >nul
|
||||||
cls
|
cls
|
||||||
|
|
||||||
set "REPO_DIR=%~dp0"
|
for %%I in ("%~dp0.") do set "REPO_DIR=%%~fI"
|
||||||
pushd "%REPO_DIR%"
|
pushd "%REPO_DIR%"
|
||||||
|
|
||||||
set "YOLO_LAUNCHER="
|
set "YOLO_MODE="
|
||||||
where yolo >nul 2>nul
|
where yolo >nul 2>nul
|
||||||
if %ERRORLEVEL% equ 0 set "YOLO_LAUNCHER=yolo"
|
if %ERRORLEVEL% equ 0 goto launcher_cli
|
||||||
|
|
||||||
if not defined YOLO_LAUNCHER (
|
py -3.11 -c "import ultralytics" >nul 2>nul
|
||||||
py -3.11 -c "import ultralytics" >nul 2>nul
|
if %ERRORLEVEL% equ 0 goto launcher_py
|
||||||
if %ERRORLEVEL% equ 0 set "YOLO_LAUNCHER=py -3.11 -m ultralytics"
|
|
||||||
)
|
|
||||||
|
|
||||||
if not defined YOLO_LAUNCHER (
|
python -c "import ultralytics" >nul 2>nul
|
||||||
python -c "import ultralytics" >nul 2>nul
|
if %ERRORLEVEL% equ 0 goto launcher_python
|
||||||
if %ERRORLEVEL% equ 0 set "YOLO_LAUNCHER=python -m ultralytics"
|
|
||||||
)
|
|
||||||
|
|
||||||
if not defined YOLO_LAUNCHER (
|
echo [ERROR] No usable Ultralytics launcher was found.
|
||||||
echo [错误] 未找到可用的 Ultralytics 启动方式
|
|
||||||
echo.
|
|
||||||
echo 请先确保满足以下任一条件:
|
|
||||||
echo 1. yolo 命令已加入 PATH
|
|
||||||
echo 2. py -3.11 可运行并已安装 ultralytics
|
|
||||||
echo 3. python 可运行并已安装 ultralytics
|
|
||||||
popd
|
|
||||||
pause
|
|
||||||
exit /b 1
|
|
||||||
)
|
|
||||||
|
|
||||||
set "DATASET=%REPO_DIR%datasets\shoe-roi-mix\data.yaml"
|
|
||||||
|
|
||||||
if not exist "%DATASET%" (
|
|
||||||
echo [错误] 找不到 ROI 数据集配置: %DATASET%
|
|
||||||
echo.
|
|
||||||
echo 请先构建 ROI 数据集:
|
|
||||||
echo python "%REPO_DIR%09_build_roi_shoe_dataset.py" --clean
|
|
||||||
popd
|
|
||||||
pause
|
|
||||||
exit /b 1
|
|
||||||
)
|
|
||||||
|
|
||||||
echo ============================================================
|
|
||||||
echo 训练鞋子 ROI 检测模型 (YOLOv8s + 640x640)
|
|
||||||
echo ============================================================
|
|
||||||
echo.
|
|
||||||
echo [信息] 数据集: %DATASET%
|
|
||||||
echo [信息] 模型: yolov8s.pt
|
|
||||||
echo.
|
echo.
|
||||||
|
echo Make sure one of the following works:
|
||||||
|
echo 1. yolo is in PATH
|
||||||
|
echo 2. py -3.11 can import ultralytics
|
||||||
|
echo 3. python can import ultralytics
|
||||||
|
goto fail
|
||||||
|
|
||||||
|
:launcher_cli
|
||||||
|
set "YOLO_MODE=cli"
|
||||||
|
goto launcher_done
|
||||||
|
|
||||||
|
:launcher_py
|
||||||
|
set "YOLO_MODE=py"
|
||||||
|
goto launcher_done
|
||||||
|
|
||||||
|
:launcher_python
|
||||||
|
set "YOLO_MODE=python"
|
||||||
|
goto launcher_done
|
||||||
|
|
||||||
|
:launcher_done
|
||||||
|
set "DATASET=%REPO_DIR%\datasets\shoe-roi-mix\data.yaml"
|
||||||
|
set "MODEL=yolov8s.pt"
|
||||||
set "EPOCHS=150"
|
set "EPOCHS=150"
|
||||||
set "IMGSZ=640"
|
set "IMGSZ=640"
|
||||||
set "BATCH=16"
|
set "BATCH=16"
|
||||||
set "PROJECT=%REPO_DIR%runs\roi_yolov8s_640"
|
set "PROJECT=%REPO_DIR%\runs\roi_yolov8s_640"
|
||||||
set "RUN_NAME=train_roi"
|
set "RUN_NAME=train_roi"
|
||||||
|
|
||||||
echo 训练参数:
|
if defined DATASET_OVERRIDE set "DATASET=%DATASET_OVERRIDE%"
|
||||||
echo - Epochs: %EPOCHS%
|
if defined MODEL_OVERRIDE set "MODEL=%MODEL_OVERRIDE%"
|
||||||
echo - Image Size: %IMGSZ%x%IMGSZ%
|
if defined EPOCHS_OVERRIDE set "EPOCHS=%EPOCHS_OVERRIDE%"
|
||||||
echo - Batch Size: %BATCH%
|
if defined IMGSZ_OVERRIDE set "IMGSZ=%IMGSZ_OVERRIDE%"
|
||||||
echo - Device: GPU (cuda:0)
|
if defined BATCH_OVERRIDE set "BATCH=%BATCH_OVERRIDE%"
|
||||||
echo - Project: %PROJECT%
|
if defined PROJECT_OVERRIDE set "PROJECT=%PROJECT_OVERRIDE%"
|
||||||
echo - Run Name: %RUN_NAME% ^(已存在时会自动递增，不覆盖旧模型^)
|
if defined RUN_NAME_OVERRIDE set "RUN_NAME=%RUN_NAME_OVERRIDE%"
|
||||||
echo.
|
|
||||||
|
|
||||||
echo ============================================================
|
if not exist "%DATASET%" (
|
||||||
echo 开始训练
|
echo [ERROR] ROI dataset yaml not found: %DATASET%
|
||||||
echo ============================================================
|
|
||||||
echo.
|
|
||||||
|
|
||||||
call %YOLO_LAUNCHER% detect train data="%DATASET%" model="yolov8s.pt" epochs=%EPOCHS% imgsz=%IMGSZ% batch=%BATCH% device=0 project="%PROJECT%" name="%RUN_NAME%"
|
|
||||||
|
|
||||||
if %ERRORLEVEL% neq 0 (
|
|
||||||
echo.
|
echo.
|
||||||
echo [错误] 训练失败！
|
echo Build the ROI dataset first:
|
||||||
popd
|
echo python "%REPO_DIR%\09_build_roi_shoe_dataset.py" --clean
|
||||||
pause
|
goto fail
|
||||||
exit /b 1
|
|
||||||
)
|
)
|
||||||
|
|
||||||
echo.
|
|
||||||
echo ============================================================
|
echo ============================================================
|
||||||
echo 训练完成！
|
echo Train ROI Shoe Detector ^(YOLOv8s + 640x640^)
|
||||||
echo ============================================================
|
echo ============================================================
|
||||||
echo.
|
echo.
|
||||||
echo 模型输出目录: %PROJECT%
|
echo [INFO] Repo: %REPO_DIR%
|
||||||
|
echo [INFO] Dataset: %DATASET%
|
||||||
|
echo [INFO] Model: %MODEL%
|
||||||
|
echo [INFO] Launcher: %YOLO_MODE%
|
||||||
|
echo.
|
||||||
|
echo Training params:
|
||||||
|
echo Epochs: %EPOCHS%
|
||||||
|
echo Image Size: %IMGSZ%x%IMGSZ%
|
||||||
|
echo Batch Size: %BATCH%
|
||||||
|
echo Device: GPU ^(cuda:0^)
|
||||||
|
echo Project: %PROJECT%
|
||||||
|
echo Run Name: %RUN_NAME%
|
||||||
|
echo.
|
||||||
|
|
||||||
|
if defined DRY_RUN (
|
||||||
|
echo [INFO] DRY_RUN is set. Exiting before training.
|
||||||
|
goto success
|
||||||
|
)
|
||||||
|
|
||||||
|
if "%YOLO_MODE%"=="cli" goto run_cli
|
||||||
|
if "%YOLO_MODE%"=="py" goto run_py
|
||||||
|
if "%YOLO_MODE%"=="python" goto run_python
|
||||||
|
|
||||||
|
echo [ERROR] Internal launcher selection error.
|
||||||
|
goto fail
|
||||||
|
|
||||||
|
:run_cli
|
||||||
|
yolo detect train data="%DATASET%" model="%MODEL%" epochs=%EPOCHS% imgsz=%IMGSZ% batch=%BATCH% device=0 project="%PROJECT%" name="%RUN_NAME%"
|
||||||
|
goto check_result
|
||||||
|
|
||||||
|
:run_py
|
||||||
|
py -3.11 -m ultralytics detect train data="%DATASET%" model="%MODEL%" epochs=%EPOCHS% imgsz=%IMGSZ% batch=%BATCH% device=0 project="%PROJECT%" name="%RUN_NAME%"
|
||||||
|
goto check_result
|
||||||
|
|
||||||
|
:run_python
|
||||||
|
python -m ultralytics detect train data="%DATASET%" model="%MODEL%" epochs=%EPOCHS% imgsz=%IMGSZ% batch=%BATCH% device=0 project="%PROJECT%" name="%RUN_NAME%"
|
||||||
|
goto check_result
|
||||||
|
|
||||||
|
:check_result
|
||||||
|
if %ERRORLEVEL% neq 0 (
|
||||||
|
echo.
|
||||||
|
echo [ERROR] Training failed.
|
||||||
|
goto fail
|
||||||
|
)
|
||||||
|
|
||||||
|
:success
|
||||||
|
echo.
|
||||||
|
echo ============================================================
|
||||||
|
echo Training Complete
|
||||||
|
echo ============================================================
|
||||||
|
echo.
|
||||||
|
echo Output project: %PROJECT%
|
||||||
|
echo.
|
||||||
|
popd
|
||||||
|
endlocal
|
||||||
|
exit /b 0
|
||||||
|
|
||||||
|
:fail
|
||||||
echo.
|
echo.
|
||||||
popd
|
popd
|
||||||
pause
|
pause
|
||||||
|
endlocal
|
||||||
|
exit /b 1
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user