代码语言:javascript复制
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()