opencv-python 车牌检测和识别
阅读原文时间:2023年08月11日阅读:4

首先利用级联分类器把车牌位置找到取出来,然后用ocr进行车牌识别。

1 OCR之Tesseract安装

Tesseract安装可以参考这个链接: https://blog.csdn.net/m0_53192838/article/details/127432761

写的比较详细,我在使用的时候有一个报错有点意思,记录一下:

Tesseract ocr识别报错:tesseract is not installed or it‘s not in your PATH

已经安装好tesseract,python中的pytesseract也装好了,但是代码里面用pytesseract的时候报错了,看起来是路径有问题,但是环境变量是已经配置好了的。

网上分析是pytesseract.py文件里的路径有问题。

把文件中的路径 tesseract_cmd = ‘tesseract’ 改为本地的绝对路径就行

tesseract_cmd = r ’D\Tesseract_ocr\tesseract.exe’

2 车牌检测和识别

网上找了三幅带有车牌的图片,肉眼看起来都比较清晰,作为此次车牌识别的原图。

 

首先用opencv的级联分类器把车牌提取出来,然后可以进行适当的形态学操作方便ocr更好的识别,然后直接用pytesseract进行识别就行了。

import cv2
import numpy as np
import pytesseract

#创建级联分类器
car = cv2.CascadeClassifier('./haarcascade_russian_plate_number.xml')

img = cv2.imread('./car1.jpg')
gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)

kernel = cv2.getStructuringElement(cv2.MORPH_RECT,(3,3)) #形态学结构元

car_nums = car.detectMultiScale(gray) #车牌检测(检测出来的框偏大)

for car_num in car_nums:
(x,y,w,h) = car_num
cv2.rectangle(img,(x,y),(x+w,y+h),[0,0,255],2) #用矩形把车牌框起来
roi = gray[y:y+h,x:x+w] #把车牌图片提取出来

\_,roi\_thresh = cv2.threshold(roi,0,255,cv2.THRESH\_BINARY | cv2.THRESH\_OTSU) #对提取的车牌二值化

open\_img = cv2.morphologyEx(roi\_thresh,cv2.MORPH\_OPEN,kernel)  #形态学开操作(二值化后效果还是差点)

cv2.imshow('open\_img',open\_img)

print(pytesseract.image\_to\_string(open\_img,lang='chi\_sim+eng',config='--psm 7 --oem 3')) # 进行ocr识别

cv2.imshow('car',img)

cv2.waitKey(0)
cv2.destroyAllWindows() 

识别结果如下:

  

 

总体来说,效果都不是很好(第三个车牌识别失败),还可以进行其他形态学操作尝试,还有部分原因是车牌框太大了,干扰比较多。

通过观察,车牌的颜色都比较固定,且和车身差距比较大,可以用之前的颜色mask方法来提取车牌(形态学操作--小狗提取优化),然后进行适当形态学操作,再进行轮廓和轮廓外接矩形的计算,然后再提取车牌,这样提取的车牌框应该会更准确一些。

import cv2
import numpy as np
import pytesseract

img = cv2.imread('./car2.jpg')
gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)

print(img.shape)

HSV = cv2.cvtColor(img, cv2.COLOR_BGR2HSV) #转换到hsv空间

kernel = cv2.getStructuringElement(cv2.MORPH_RECT,(3,3)) #形态学结构元

LowerBlue = np.array([90, 190, 100]) #检测hsv的上下限(蓝色车牌)
UpperBlue = np.array([130, 230, 200])

#inRange 函数将颜色的值设置为 1,如果颜色存在于给定的颜色范围内,则设置为白色,如果颜色不存在于指定的颜色范围内,则设置为 0
mask = cv2.inRange(HSV,LowerBlue,UpperBlue) #车牌mask
cv2.imshow('mask',mask)

dilate = cv2.morphologyEx(mask,cv2.MORPH_DILATE,kernel,iterations=4) #形态学膨胀和开操作把提取的蓝色点连接起来
morph = cv2.morphologyEx(dilate,cv2.MORPH_CLOSE,kernel,iterations=6)
cv2.imshow('morph',morph)

_,contours,_ = cv2.findContours(morph,cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_SIMPLE) #找车牌的轮廓,只找外轮廓就行

print(len(contours))

img_copy = img.copy()
cv2.drawContours(img_copy,contours,-1,[0,0,255],2) #把轮廓画出来
cv2.imshow('img_copy',img_copy)

rect = cv2.boundingRect(contours[0]) #用矩形把轮廓框出来(轮廓外接矩形)
(x,y,w,h) = rect
cv2.rectangle(img,(x,y),(x+w,y+h),[0,0,255],2)
cv2.imshow('car',img)

roi_img = gray[y:y+h,x:x+w] #提取车牌区域进行ocr识别

_,roi_thresh = cv2.threshold(roi_img,0,255,cv2.THRESH_BINARY | cv2.THRESH_OTSU)

open_img = cv2.morphologyEx(roi_thresh,cv2.MORPH_OPEN,kernel) #适当的形态学操作提高识别率

cv2.imshow('open_img',open_img)

print(pytesseract.image_to_string(roi_img,lang='chi_sim+eng',config='--psm 8 --oem 3')) #ocr识别

cv2.waitKey(0)
cv2.destroyAllWindows()

 

 

 

车牌框比较准确,识别效果相对好了不少,针对具体问题还可以继续微调形态学和 inRange 的范围参数,提高识别率。