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:
20
build_int8_480.bat
Normal file
20
build_int8_480.bat
Normal file
@@ -0,0 +1,20 @@
|
||||
@echo off
|
||||
chcp 65001 >nul
|
||||
echo ============================================
|
||||
echo Building INT8 480p TensorRT Engine
|
||||
echo ============================================
|
||||
echo.
|
||||
|
||||
call conda activate yolo
|
||||
|
||||
echo Building engine...
|
||||
trtexec --onnx=yolo11n.onnx --saveEngine=yolo11n_int8_480.engine --explicitBatch --int8 --fp16 --workspace=4096 --builderOptimizationLevel=4 --profilingVerbosity=detailed --calib=yolo11n_int8_480.cache --optShapes=input:4x3x480x480 --maxShapes=input:8x3x480x480 --minShapes=input:1x3x480x480 --useCudaGraph --useSpinWait --noTF32
|
||||
|
||||
if %errorlevel% equ 0 (
|
||||
echo.
|
||||
echo [SUCCESS] INT8 480p Engine: yolo11n_int8_480.engine
|
||||
) else (
|
||||
echo [FAILED] INT8 480p Engine build failed
|
||||
)
|
||||
|
||||
pause
|
||||
Reference in New Issue
Block a user