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

67
build_engine.bat Normal file
View File

@@ -0,0 +1,67 @@
@echo off
chcp 65001 >nul
echo ============================================
echo YOLO INT8 TensorRT Engine Builder
echo ============================================
echo.
REM 激活虚拟环境
echo Activating yolo virtual environment...
call conda activate yolo
echo.
REM 检查ONNX模型
if not exist yolo11n.onnx (
echo ERROR: yolo11n.onnx not found!
echo Please export the ONNX model first.
pause
exit /b 1
)
REM 检查校准缓存
if not exist yolo11n_int8.cache (
echo ERROR: yolo11n_int8.cache not found!
echo Please run calibration_gen.py first to generate the calibration cache.
pause
exit /b 1
)
echo Building TensorRT Engine with INT8 calibration...
echo ONNX: yolo11n.onnx
echo Engine: yolo11n_int8_b1_8.engine
echo Cache: yolo11n_int8.cache
echo.
trtexec ^
--onnx=yolo11n.onnx ^
--saveEngine=yolo11n_int8_b1_8.engine ^
--explicitBatch ^
--int8 ^
--fp16 ^
--workspace=4096 ^
--builderOptimizationLevel=5 ^
--profilingVerbosity=detailed ^
--calib=yolo11n_int8.cache ^
--minShapes=input:1x3x480x640 ^
--optShapes=input:4x3x640x640 ^
--maxShapes=input:8x3x640x736 ^
--useCudaGraph ^
--useSpinWait ^
--noTF32
if %errorlevel% equ 0 (
echo.
echo ============================================
echo Engine built successfully!
echo Output: yolo11n_int8_b1_8.engine
echo ============================================
) else (
echo.
echo ============================================
echo Engine build failed!
echo ============================================
pause
exit /b 1
)
pause