43 lines
1.1 KiB
Batchfile
43 lines
1.1 KiB
Batchfile
|
@echo off
|
||
|
|
chcp 65001 >nul
|
||
|
|
echo ============================================
|
||
|
|
echo YOLO INT8 TensorRT Validation Script
|
||
|
|
echo ============================================
|
||
|
|
echo.
|
||
|
|
|
||
|
|
REM 激活虚拟环境
|
||
|
|
echo Activating yolo virtual environment...
|
||
|
|
call conda activate yolo
|
||
|
|
echo.
|
||
|
|
|
||
|
|
REM 检查engine文件
|
||
|
|
if not exist yolo11n_int8_b1_8.engine (
|
||
|
|
echo ERROR: yolo11n_int8_b1_8.engine not found!
|
||
|
|
echo Please build the engine first using: build_engine.bat
|
||
|
|
pause
|
||
|
|
exit /b 1
|
||
|
|
)
|
||
|
|
|
||
|
|
echo Validating INT8 TensorRT Engine with COCO dataset...
|
||
|
|
echo Engine: yolo11n_int8_b1_8.engine
|
||
|
|
echo.
|
||
|
|
|
||
|
|
REM 使用rect=False确保输入尺寸为640x640
|
||
|
|
yolo val model=yolo11n_int8_b1_8.engine data=coco.yaml imgsz=640 rect=False
|
||
|
|
|
||
|
|
if %errorlevel% equ 0 (
|
||
|
|
echo.
|
||
|
|
echo ============================================
|
||
|
|
echo Validation completed successfully!
|
||
|
|
echo ============================================
|
||
|
|
) else (
|
||
|
|
echo.
|
||
|
|
echo ============================================
|
||
|
|
echo Validation failed!
|
||
|
|
echo ============================================
|
||
|
|
pause
|
||
|
|
exit /b 1
|
||
|
|
)
|
||
|
|
|
||
|
|
pause
|