Function main

src-tauri/src-python/scripts/test_activeness_ocr.py:853–886

Execute main script logic to test OCR performance on debug frames.

()

def main():
    """Execute main script logic to test OCR performance on debug frames."""
    rapid = RapidOCRBackend.pp_ocr_v5_rec()

    # Try to load Qwen2-VL for rankings (mirrors production _select_ocr_backend)
    qwen = None
    if QwenVLOCRBackend.has_sufficient_vram():
        print("[rankings] GPU detected — using Qwen2-VL-2B for DR/SA rankings.")
        qwen = QwenVLOCRBackend()
    else:
        print(
            "[rankings] No GPU / insufficient VRAM — using RapidOCR for DR/SA rankings."
        )

    test_activeness(rapid)
    test_chest_contributions(rapid)

    # Find all ranking prefixes dynamically (dr_*, sa_*)
    dr_prefixes = sorted(
        {"_".join(p.name.split("_")[:2]) for p in SCREENSHOTS_DIR.glob("dr_*.png")}
    )
    sa_prefixes = sorted(
        {"_".join(p.name.split("_")[:2]) for p in SCREENSHOTS_DIR.glob("sa_*.png")}
    )

    for prefix in dr_prefixes:
        test_rankings(
            rapid, qwen, prefix, f"DREAM REALM — {prefix}", is_supreme_arena=False
        )

    for prefix in sa_prefixes:
        test_rankings(
            rapid, qwen, prefix, f"SUPREME ARENA — {prefix}", is_supreme_arena=True
        )


if __name__ == "__main__":