16337
942244bd88
- Engine build scripts (FP16/INT8) - Benchmark validation scripts - Result parsing and analysis tools - COCO dataset configuration
34 lines
1.7 KiB
Batchfile
34 lines
1.7 KiB
Batchfile
@echo off
|
|
echo ============================================
|
|
echo Rebuilding TensorRT engines with batch 1-8
|
|
echo ============================================
|
|
|
|
call conda activate yolo
|
|
|
|
echo.
|
|
echo [1/5] Building INT8 640p (batch 1-8)...
|
|
trtexec --onnx=yolo11n.onnx --saveEngine=yolo11n_int8_b1_8.engine --explicitBatch --int8 --fp16 --workspace=4096 --builderOptimizationLevel=4 --calib=yolo11n_int8.cache --minShapes=input:1x3x640x640 --optShapes=input:4x3x640x640 --maxShapes=input:8x3x640x640 --useCudaGraph --useSpinWait --noTF32
|
|
echo Done.
|
|
|
|
echo.
|
|
echo [2/5] Building FP16 640p (batch 1-8)...
|
|
trtexec --onnx=yolo11n.onnx --saveEngine=yolo11n_fp16_640.engine --explicitBatch --fp16 --workspace=4096 --builderOptimizationLevel=4 --minShapes=input:1x3x640x640 --optShapes=input:4x3x640x640 --maxShapes=input:8x3x640x640 --useCudaGraph --useSpinWait --noTF32
|
|
echo Done.
|
|
|
|
echo.
|
|
echo [3/5] Building INT8 480p (batch 1-8)...
|
|
trtexec --onnx=yolo11n.onnx --saveEngine=yolo11n_int8_480.engine --explicitBatch --int8 --fp16 --workspace=4096 --builderOptimizationLevel=4 --calib=yolo11n_int8_480.cache --minShapes=input:1x3x480x480 --optShapes=input:4x3x480x480 --maxShapes=input:8x3x480x480 --useCudaGraph --useSpinWait --noTF32
|
|
echo Done.
|
|
|
|
echo.
|
|
echo [4/5] Building FP16 480p (batch 1-8)...
|
|
trtexec --onnx=yolo11n.onnx --saveEngine=yolo11n_fp16_480.engine --explicitBatch --fp16 --workspace=4096 --builderOptimizationLevel=4 --minShapes=input:1x3x480x480 --optShapes=input:4x3x480x480 --maxShapes=input:8x3x480x480 --useCudaGraph --useSpinWait --noTF32
|
|
echo Done.
|
|
|
|
echo.
|
|
echo ============================================
|
|
echo All engines rebuilt!
|
|
echo Now run: run_all_benchmarks.bat
|
|
echo ============================================
|
|
pause
|