python识别车牌号

爱被打了一巴掌 2022-09-02 04:28 249阅读 0赞

import cv2
    
    """定义一些参数"""
    # imshow 窗口的尺寸
    frameWidth = 640
    frameHeight = 480
    # 导入 xml 文件（根据自己的路径进行相对应的调整）
    numberPlatesCascade = cv2.CascadeClassifier("C:/Users/geek/Desktop/haarcascade_russian_plate_number.xml")
    # 设定一个可以被检测到的最小的物体的面积，可用于去除一些不必要的噪声
    minArea = 500
    # bounding box 的颜色
    color = (255, 0, 255)
    
    # 导入计算机自带的摄像头，并设置摄像头所拍摄画面的尺寸、亮度
    cap = cv2.VideoCapture(0)
    cap.set(3, frameWidth)
    cap.set(4, frameHeight)
    cap.set(10, 150)
    # 用于保存车牌照片时的计数
    count = 0
    global imgRoi
    
    while True:
        success, img = cap.read()
        imgGray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)  # 转为灰度图像
        numberPlates = numberPlatesCascade.detectMultiScale(imgGray, 1.1, 4)  # 调用xml文件抓到图像中的车牌
    
        for (x, y, w, h) in numberPlates:
            area = w * h
            if area > minArea:
                cv2.rectangle(img, (x, y), (x + w, y + h), color, 2)  # 添加 bounding box
                cv2.putText(img, "NUMBER PLATE", (x, y-5),  
                            cv2.FONT_HERSHEY_PLAIN, 1, color, 2)  # 给 bounding box 添加注解
            # 单独将车牌抓出来，另外显示
            imgRoi = img[y:y + h + 30, x:x + w]
            cv2.imshow("Number Plate", imgRoi)
    
        cv2.imshow("Result", img)
        k=cv2.waitKey(1)
        if k & 0xFF == ord("s") and 'imgRoi' in dir():  # 用于保存抓到的车牌
            cv2.imwrite("C:/Users/geek/Desktop/platenum/NumberPlate_" + str(count) + ".jpg", imgRoi)
            cv2.rectangle(img, (0, 200), (640, 300), (0, 255, 0), cv2.FILLED)  # 做一个用于提示保存成功地提示条
            cv2.putText(img, "Scan Saved", (150, 265), cv2.FONT_HERSHEY_PLAIN,
                        2, (0, 255, 255), 2)  # 在提示条上写上内容
            cv2.imshow("Result", img)
            cv2.waitKey(500)
            count += 1
        if k == 27:
            #通过esc键退出摄像
            cv2.destroyAllWindows()
            break
    #关闭摄像头
    cap.release()