OCR_NUMBERPLATE_YOLO / __init__.py
Pujan-Dev's picture
test
f9003ec
"""
OCR Number Plate Pipeline
=========================
A complete pipeline for detecting and recognizing number plates,
supporting both Nepali and English characters.
Components:
- PlateDetector: YOLO-based number plate detection
- CharacterRecognizer: ResNet18-based OCR model
- NumberPlateOCR: Complete pipeline combining detection and recognition
Usage:
from pipeline import NumberPlateOCR
# Initialize pipeline
ocr = NumberPlateOCR(use_yolo=True)
# Process image
result = ocr.process_image('plate.jpg')
print(result['plates'][0]['singleline_text'])
"""
from .main import NumberPlateOCR
from .model.ocr import CharacterRecognizer, OCRModel
from .model.plate_detector import PlateDetector, PlateDetectorLite, get_detector
from .config.config import (
OCR_CONFIG, YOLO_CONFIG, CONTOUR_CONFIG,
get_device, setup_directories
)
__version__ = "1.0.0"
__author__ = "OCR Team"
__all__ = [
'NumberPlateOCR',
'CharacterRecognizer',
'OCRModel',
'PlateDetector',
'PlateDetectorLite',
'get_detector',
'OCR_CONFIG',
'YOLO_CONFIG',
'CONTOUR_CONFIG',
'get_device',
'setup_directories'
]