Python-opencv 人脸检测并添加方框或圆框

本博客运行环境为树莓派，python 3.7 。要提前安装好opencv库。

使用opencv库中的haarcascade_frontalface_default.xml进行人脸检测，在相应的位置处调用函数加上方框框或圆框框。并能把视频保存下来。
原来的视频命名为output.avi，代码编译后保存的视频命名为111.avi

代码如下：

import cv2 as cv

cap = cv.VideoCapture('output.avi')
face_cascade = cv.CascadeClassifier("/usr/share/opencv/haarcascades/haarcascade_frontalface_default.xml")

fourcc = cv.VideoWriter_fourcc(*'XVID')
out = cv.VideoWriter('111.avi',fourcc, 20.0, (640,480))

#eye_cascade = cv.CascadeClassifier("/usr/share/opencv/haarcascades/haarcascade_eye.xml") #眼睛
def facedetection(img):
    faces = face_cascade.detectMultiScale(img, 1.3, 5)
    for (x, y, w, h) in faces:
        #方框
        # img = cv2.rectangle(img, (x, y), (x+w, y+h), (255, 0, 0), 2)
        # roi_gray = gray[y:y+h, x:x+w]

        #圆框
        cv.circle(img, (int(x+w/2), int(y+h/2)), 100, (0, 0, 255), 2)
    return img

while True:
    # 读取当前帧
    ret, frame = cap.read()
    # 转为灰度图像
    gray = cv.cvtColor(frame, cv.COLOR_BGR2GRAY)
    im = facedetection(frame)
    out.write(im)
    
cap.release()
out.release()
cv2.destroyAllWindows()