@echo off
chcp 65001 >nul
echo ============================================
echo Building INT8 480p TensorRT Engine
echo ============================================
echo.

call conda activate yolo

echo Building engine...
trtexec --onnx=yolo11n.onnx --saveEngine=yolo11n_int8_480.engine --explicitBatch --int8 --fp16 --workspace=4096 --builderOptimizationLevel=4 --profilingVerbosity=detailed --calib=yolo11n_int8_480.cache --optShapes=input:4x3x480x480 --maxShapes=input:8x3x480x480 --minShapes=input:1x3x480x480 --useCudaGraph --useSpinWait --noTF32

if %errorlevel% equ 0 (
    echo.
    echo [SUCCESS] INT8 480p Engine: yolo11n_int8_480.engine
) else (
    echo [FAILED] INT8 480p Engine build failed
)

pause