@echo off
chcp 65001 >nul
echo ============================================
echo YOLO INT8 TensorRT Validation Script
echo ============================================
echo.

REM 激活虚拟环境
echo Activating yolo virtual environment...
call conda activate yolo
echo.

REM 检查engine文件
if not exist yolo11n_int8_b1_8.engine (
    echo ERROR: yolo11n_int8_b1_8.engine not found!
    echo Please build the engine first using: build_engine.bat
    pause
    exit /b 1
)

echo Validating INT8 TensorRT Engine with COCO dataset...
echo Engine: yolo11n_int8_b1_8.engine
echo.

REM 使用rect=False确保输入尺寸为640x640
yolo val model=yolo11n_int8_b1_8.engine data=coco.yaml imgsz=640 rect=False

if %errorlevel% equ 0 (
    echo.
    echo ============================================
    echo Validation completed successfully!
    echo ============================================
) else (
    echo.
    echo ============================================
    echo Validation failed!
    echo ============================================
    pause
    exit /b 1
)

pause