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

29
run_all_benchmarks.bat Normal file
View File

@@ -0,0 +1,29 @@
@echo off
echo ============================================
echo YOLO11n Person and Vehicle Detection Benchmark
echo ============================================
call conda activate yolo
echo [1/4] Testing FP32...
yolo val model=yolo11n.pt data=coco_person_vehicle.yaml imgsz=640 rect=False batch=1 > fp32_results.txt 2>&1
echo Done: fp32_results.txt
echo [2/4] Testing INT8 640p...
yolo val model=yolo11n_int8_b1_8.engine data=coco_person_vehicle.yaml imgsz=640 rect=False batch=1 > int8_640_results.txt 2>&1
echo Done: int8_640_results.txt
echo [3/4] Testing FP16 640p...
yolo val model=yolo11n_fp16_640.engine data=coco_person_vehicle.yaml imgsz=640 rect=False batch=1 > fp16_640_results.txt 2>&1
echo Done: fp16_640_results.txt
echo [4/4] Testing FP16 480p...
yolo val model=yolo11n_fp16_480.engine data=coco_person_vehicle.yaml imgsz=480 rect=False batch=1 > fp16_480_results.txt 2>&1
echo Done: fp16_480_results.txt
echo.
echo ============================================
echo All validation tests completed!
echo Now run: python parse_results.py
echo ============================================
pause