Add YOLO11 TensorRT quantization benchmark scripts

- Engine build scripts (FP16/INT8)
- Benchmark validation scripts
- Result parsing and analysis tools
- COCO dataset configuration
This commit is contained in:
2026-01-29 13:59:42 +08:00
commit 942244bd88
34 changed files with 3514 additions and 0 deletions

42
validate_int8.bat Normal file
View File

@@ -0,0 +1,42 @@
@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