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:
34
run_correct_benchmark.bat
Normal file
34
run_correct_benchmark.bat
Normal file
@@ -0,0 +1,34 @@
|
||||
@echo off
|
||||
echo ============================================
|
||||
echo YOLO11n Benchmark - Person & Vehicle Only
|
||||
echo Classes: 0(person),1(bicycle),2(car),3(motorcycle),5(bus),7(truck)
|
||||
echo ============================================
|
||||
echo.
|
||||
|
||||
call conda activate yolo
|
||||
|
||||
echo [1/4] Testing FP32 (PyTorch)...
|
||||
yolo val model=yolo11n.pt data=coco.yaml imgsz=640 batch=1 device=0 classes=0,1,2,3,5,7 > fp32_results.txt 2>&1
|
||||
echo Done: fp32_results.txt
|
||||
|
||||
echo.
|
||||
echo [2/4] Testing INT8 640p...
|
||||
yolo val model=yolo11n_int8_b1_8.engine data=coco.yaml imgsz=640 batch=1 device=0 classes=0,1,2,3,5,7 > int8_640_results.txt 2>&1
|
||||
echo Done: int8_640_results.txt
|
||||
|
||||
echo.
|
||||
echo [3/4] Testing FP16 640p...
|
||||
yolo val model=yolo11n_fp16_640.engine data=coco.yaml imgsz=640 batch=1 device=0 classes=0,1,2,3,5,7 > fp16_640_results.txt 2>&1
|
||||
echo Done: fp16_640_results.txt
|
||||
|
||||
echo.
|
||||
echo [4/4] Testing FP16 480p...
|
||||
yolo val model=yolo11n_fp16_480.engine data=coco.yaml imgsz=480 batch=1 device=0 classes=0,1,2,3,5,7 > fp16_480_results.txt 2>&1
|
||||
echo Done: fp16_480_results.txt
|
||||
|
||||
echo.
|
||||
echo ============================================
|
||||
echo All validation tests completed!
|
||||
echo Now run: python parse_final.py
|
||||
echo ============================================
|
||||
pause
|
||||
Reference in New Issue
Block a user