9.利用opencv训练出来的模型识别指定物体

9.利用opencv训练出来的模型识别指定物体_第1张图片
上节课我们讲到了如何训练自己的模型,当前课程我们利用训练好的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_GRAYSCALE
    )
    for (x, y, w, h) in rect:
        cv2.rectangle(frame, (x, y), (x+w, y+h), (0, 255, 0), 2)

    cv2.imshow('frame', frame)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break
cap.release()
cv2.destroyAllWindows()

执行结果如下:
9.利用opencv训练出来的模型识别指定物体_第2张图片

你可能感兴趣的:(计算机视觉(完结),图像识别,计算机视觉,opencv)