|
|
|
|
|
""" |
|
|
Quick OCR test — verify the ONNX engine works on sample images. |
|
|
|
|
|
Usage: |
|
|
python tools/test_quick.py # test all images in working_space/input/ |
|
|
python tools/test_quick.py path/to/image.png # test single image |
|
|
""" |
|
|
|
|
|
import sys |
|
|
import time |
|
|
from pathlib import Path |
|
|
|
|
|
from PIL import Image |
|
|
|
|
|
sys.path.insert(0, str(Path(__file__).parent.parent)) |
|
|
from ocr.engine_onnx import OcrEngineOnnx |
|
|
|
|
|
|
|
|
def test_image(engine: OcrEngineOnnx, path: Path) -> bool: |
|
|
"""Run OCR on one image and print results.""" |
|
|
img = Image.open(path) |
|
|
t0 = time.perf_counter() |
|
|
result = engine.recognize_pil(img) |
|
|
elapsed = (time.perf_counter() - t0) * 1000 |
|
|
|
|
|
status = "OK" if result.text.strip() else "EMPTY" |
|
|
text_short = result.text.replace("\n", " | ")[:80] |
|
|
|
|
|
print(f" [{status}] {path.name:30s} {elapsed:6.0f}ms " |
|
|
f"L={len(result.lines):2d} C={result.average_confidence:.0%} " |
|
|
f'"{text_short}"') |
|
|
return bool(result.text.strip()) |
|
|
|
|
|
|
|
|
def main(): |
|
|
if len(sys.argv) > 1: |
|
|
paths = [Path(p) for p in sys.argv[1:]] |
|
|
else: |
|
|
input_dir = Path("working_space/input") |
|
|
if not input_dir.exists(): |
|
|
print(f"No images found. Place .png files in {input_dir}") |
|
|
return |
|
|
paths = sorted(input_dir.glob("*.png")) |
|
|
|
|
|
if not paths: |
|
|
print("No images to test.") |
|
|
return |
|
|
|
|
|
print(f"Testing {len(paths)} image(s)...\n") |
|
|
engine = OcrEngineOnnx() |
|
|
|
|
|
ok = sum(test_image(engine, p) for p in paths) |
|
|
print(f"\n Result: {ok}/{len(paths)} images produced text") |
|
|
|
|
|
|
|
|
if __name__ == "__main__": |
|
|
main() |
|
|
|
