OpenCV-Python实现实时车牌检测(附源码)

OpenCV杂谈_02


一. 需要做的前期准备

  1. 环境配置
    Python版本: 3.7.6
    用到的功能包: opencv-python (4.2.0.32)
  2. 导入他人训练好的 OpenCV cascade 文件
    haarcascade_russian_plate_number.xml (也可以用其他.xml文件用于做不同的检测,比如全身、人眼等等)下载链接在此:https://github.com/opencv/opencv/tree/master/data/haarcascades
  3. 提前创建好用于保存抓到的车牌的文件夹(作者这里将其命名为“Scanned”)
  4. 一个用的顺手的IDE(本人推荐Pycharm)

二. 源码如下(需要做路径上的调整)

import cv2

"""定义一些参数"""
# imshow 窗口的尺寸
frameWidth = 640
frameHeight = 480
# 导入 xml 文件(根据自己的路径进行相对应的调整)
numberPlatesCascade = cv2.CascadeClassifier("Resources/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

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)
    if cv2.waitKey(1) & 0xFF == ord("s"):  # 用于保存抓到的车牌
        cv2.imwrite("Resources/Scanned/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

三. 结果展示

四. 教程推荐

个人觉得OpenCV不错的教程:https://www.bilibili.com/video/BV16K411W7x9?t=6282


如有问题,敬请指正。欢迎转载,但请注明出处。

你可能感兴趣的:(OpenCV自学整理,opencv,python)