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:
42
validate_int8.bat
Normal file
42
validate_int8.bat
Normal 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
|
||||
Reference in New Issue
Block a user