main.py

import os
import glob
import cv2
import easyocr
from Utils_NumberPlateRecognition import check1Line, corrrectCity, correctCapital, checkCity, checkOddEven, correctNumber, showText
import Utils

imagesPath = "Images\\"
reader = easyocr.Reader(['en'])
numberPlatePoints = []
heightNumberPlate = 250

def main(dirFile):
    # for filename in glob.glob(imagesPath+"*.jpg"):
    img = cv2.imread(dirFile)
    scale = round(heightNumberPlate / img.shape[0], 1)
    img = Utils.resize(img, scale)
    detail = reader.readtext(img)
    numberPlate, datePlate = check1Line(img, detail)
    return numberPlate
  
def detailNumberPlate(numberPlate):
    numberPlate = corrrectCity(numberPlate)
    numberPlate = correctCapital(numberPlate)
    code, city = checkCity(numberPlate)
    oddEven = checkOddEven(numberPlate)
    return numberPlate, code, city, oddEven

def detailDatePlate(datePlate):
    datePlate = correctNumber(datePlate)

    month = int(datePlate[:2])
    year = int("20" + datePlate[-2:])
    return datePlate, month, year

def printDetail(numberPlate, datePlate, code, city, oddEven, month, year):
    print("---"*20)
    print("numberPlateFix:", numberPlate)
    print("datePlateFix:", datePlate)
    print("- city code:", code)
    print("- city:", city)
    print("- type:", oddEven)
    print("- month:", month)
    print("- year:", year)
    print("---"*20)