"""Минимальная проверка нового пайплайна.""" import sys from pathlib import Path # Чтобы импорты из app/ работали при запуске из корня sys.path.insert(0, str(Path(__file__).resolve().parent.parent)) from app.services.pipeline import PlatePipeline ROOT = Path(__file__).resolve().parent.parent DET_WEIGHTS = ROOT / "models" / "yolov8_plate.pt" def main(image_path: str): print("Инициализация пайплайна...") pipeline = PlatePipeline(DET_WEIGHTS) print(f"\nОбрабатываю: {image_path}") detections = pipeline.process_file(image_path) if not detections: print("Ничего не найдено.") return for i, d in enumerate(detections, 1): marker = "✓ ГОСТ" if d["is_valid_gost"] else "✗ не по ГОСТ" fb = " (fallback на всё изображение)" if d.get("fallback") else "" print(f"\nДетекция #{i}{fb}:") print(f" bbox: {d['bbox']}") print(f" confidence: {d['confidence']:.3f}") print(f" OCR raw: {d['plate_text_raw']}") print(f" Номер: {d['plate_text']} [{marker}]") if __name__ == "__main__": if len(sys.argv) < 2: print("Использование: python scripts/test_pipeline.py <путь>") sys.exit(1) main(sys.argv[1])