Files
Test_AI/build_fp16_480.bat

23 lines
748 B
Batchfile
Raw Normal View History

@echo off
echo ============================================
echo Build & Test FP16 480p (correct way)
echo ============================================
call conda activate yolo
echo [1/3] Delete old engine...
del /Q yolo11n_fp16_480.engine 2>nul
echo Done.
echo.
echo [2/3] Build FP16 480p engine (static 1x3x480x480)...
trtexec --onnx=yolo11n.onnx --saveEngine=yolo11n_fp16_480.engine --explicitBatch --fp16 --minShapes=images:1x3x480x480 --optShapes=images:1x3x480x480 --maxShapes=images:1x3x480x480 --workspace=4096 --noTF32
echo.
echo [3/3] Test 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