准备工作
首先从GitHub(https://github.com/opencv/opencv)下载文件并解压,从data文件夹中的haarcascades复制一下文件放在自己的项目中
代码实现
# 1 导入库
import cv2
# 方法: 绘制图片中检测到的人脸
def plot_rectangle(image, faces):
# 拿到检测到的人脸数据, 返回4个值:坐标(x,y), 宽高width, height
for (x, y, w, h) in faces:
cv2.rectangle(image, (x, y), (x+w, y+h), (255, 0, 0), 3)
return image
# 4 主函数
def main():
# 读取摄像头
capture =cv2.VideoCapture(0)
# 通过OpenCV自带的方法cv2.CascadeClassifier()加载级联分类器
face_alt2 = cv2.CascadeClassifier("haarcascade_frontalface_alt2.xml")
# 判断摄像头是否正常工作
if capture.isOpened() is False:
print("Camera Error!")
while True:
# 获取每一帧
ret, frame = capture.read()
if ret:
# 灰度转换
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
# 对图像中的人脸进行检测
face_alt2_detect = face_alt2.detectMultiScale(gray)
# 绘制图片中检测到的人脸
face_alt2_result = plot_rectangle(frame.copy(), face_alt2_detect)
cv2.imshow("face detection", face_alt2_result)
if cv2.waitKey(10) & 0xFF == ord('q'):
break
capture.release()
cv2.destroyAllWindows()
# 12 主程序入口
if __name__ == '__main__':
main()