16337
942244bd88
- Engine build scripts (FP16/INT8) - Benchmark validation scripts - Result parsing and analysis tools - COCO dataset configuration
21 lines
647 B
Batchfile
21 lines
647 B
Batchfile
@echo off
|
|
chcp 65001 >nul
|
|
echo ============================================
|
|
echo Building FP16 640p TensorRT Engine
|
|
echo ============================================
|
|
echo.
|
|
|
|
call conda activate yolo
|
|
|
|
echo Building engine...
|
|
trtexec --onnx=yolo11n.onnx --saveEngine=yolo11n_fp16_640.engine --explicitBatch --fp16 --workspace=4096 --builderOptimizationLevel=4 --profilingVerbosity=detailed --optShapes=input:4x3x640x640 --maxShapes=input:8x3x640x640 --useCudaGraph --useSpinWait --noTF32
|
|
|
|
if %errorlevel% equ 0 (
|
|
echo.
|
|
echo [SUCCESS] FP16 640p Engine: yolo11n_fp16_640.engine
|
|
) else (
|
|
echo [FAILED] FP16 640p Engine build failed
|
|
)
|
|
|
|
pause
|