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:
17
test_two_engines.bat
Normal file
17
test_two_engines.bat
Normal file
@@ -0,0 +1,17 @@
|
||||
@echo off
|
||||
echo ============================================
|
||||
echo Build FP16 480p + Test FP16 480p
|
||||
echo ============================================
|
||||
|
||||
call conda activate yolo
|
||||
|
||||
echo [1/2] Building FP16 480p engine...
|
||||
trtexec --onnx=yolo11n.onnx --saveEngine=yolo11n_fp16_480.engine --explicitBatch --fp16 --workspace=4096 --builderOptimizationLevel=4 --minShapes=images:1x3x480x480 --optShapes=images:4x3x480x480 --maxShapes=images:8x3x480x480 --useCudaGraph --useSpinWait --noTF32
|
||||
|
||||
echo.
|
||||
echo [2/2] 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.
|
||||
echo Done. Run: python parse_simple.py
|
||||
pause
|
||||
Reference in New Issue
Block a user