上节课我们讲到了如何训练自己的模型,当前课程我们利用训练好的XML模型文件,来识别指定的物体。这里我们直接会用到 cascade.xml 文件。
在项目文件夹下面新建 demo.py 文件,代码如下:
import cv2
# 加载训练好的分类器
faceCascade = cv2.CascadeClassifier("cascade.xml")
faceCascade.load('/home/xmake/Desktop/work/08_demo/cascade.xml')
cap = cv2.VideoCapture(0)
while True:
ret, frame = cap.read()
img = frame.copy()
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
rect = faceCascade.detectMultiScale(
gray,
scaleFactor=1.15,
minNeighbors=3,
minSize=(3,3),
flags = cv2.IMREAD_GRAYSC