@echo off
echo ============================================
echo Rebuilding FP16 engines with batch=1 support
echo ============================================

call conda activate yolo

echo.
echo [1/3] Rebuilding 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 [2/3] Rebuilding 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 [3/3] Rebuilding 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 ============================================
echo All engines rebuilt!
echo Now run: run_correct_benchmark.bat
echo ============================================
pause