Computer_Vision.py

#Modules 

import cv2 
from matplotlib import pyplot as plt 
import numpy as np 
import imutils
import easyocr

#Number Plate 

def NP_OCR(src):
    try:
        img = cv2.imread(src)
        gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)

        bfilter = cv2.bilateralFilter(gray, 11, 17, 17) 
        edged = cv2.Canny(bfilter, 30, 200) 

        keypoints = cv2.findContours(edged.copy(), cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
        contours = imutils.grab_contours(keypoints)
        contours = sorted(contours, key=cv2.contourArea, reverse=True)[:10]

        location = None
        for contour in contours:
            approx = cv2.approxPolyDP(contour, 10, True)
            if len(approx) == 4:
                location = approx
                break

        mask = np.zeros(gray.shape, np.uint8)
        new_image = cv2.drawContours(mask, [location], 0,255, -1)
        new_image = cv2.bitwise_and(img, img, mask=mask)

        (x,y) = np.where(mask==255)
        (x1, y1) = (np.min(x), np.min(y))
        (x2, y2) = (np.max(x), np.max(y))
        cropped_image = gray[x1:x2+1, y1:y2+1]

        reader = easyocr.Reader(['en'])
        result = reader.readtext(cropped_image)

        try:
            text = result[0][-2]
            return(text)
        except: 
            return None
    except:
        print("Number Plate Not Found")
    
def GetNumberPlate():
    cam = cv2.VideoCapture(0)
    cv2.namedWindow("Camera Frame")
    while True:
        ret, frame = cam.read()
        if not ret:
            print("Failed To Open Camera")
            break 
        cv2.imshow("Number Plate Capture", frame)

        k = cv2.waitKey(1)
        if k%256 == 27: 
            print("\n--Closing Camera--")
            break 

        if k%256 == 32:
            imgname = "VehicleNumber.jpg"
            cv2.imwrite(imgname, frame)

            NumPlt = NP_OCR(imgname)
            return(NumPlt)
#End