test_ov3_simple.py

#!/usr/bin/env python3
"""Simple OV-3 test. Run: conda activate rsgen && python test_ov3_simple.py
Requires: sam3, pycocotools (pip install pycocotools if missing)."""
from pathlib import Path

from PIL import Image

from pipeline import SegEarthPipeline

if __name__ == "__main__":
    repo = Path(__file__).resolve().parent
    img_path = repo / "demo_YESeg-OPT-SAR" / "sar.png"
    if not img_path.exists():
        img_path = repo / "demo.png"
    if not img_path.exists():
        print("No demo image found")
        exit(1)

    print("Loading OV-3...")
    pipe = SegEarthPipeline(variant="OV-3", device="cuda")
    print("Running inference...")
    img = Image.open(img_path).convert("RGB")
    pred = pipe(img)
    print(f"OK shape={pred.shape} classes={pred.unique().tolist()}")