实现从摄像头中实时检测教室内的人数

import cv2
import imutils

# 加载摄像头
cap = cv2.VideoCapture(0)

# 创建人脸检测器
face_cascade = cv2.CascadeClassifier(cv2.data.haarcascades + 'haarcascade_frontalface_default.xml')

# 初始化人数计数器
num_people = 0

while True:
    # 读取摄像头数据
    ret, frame = cap.read()

    # 调整图像大小
    frame = imutils.resize(frame, width=500)

    # 转换为灰度图像
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)

    # 检测人脸
    faces = face_cascade.detectMultiScale(gray, 1.3, 5)

    # 绘制人脸矩形框
    for (x, y, w, h) in faces:
        cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 255, 0), 2)

    # 更新人数计数器
    num_people = len(faces)

    # 在图像上显示人数
    cv2.putText(frame, "Number of People: {}".format(num_people), (10, 30), cv2.FONT_HERSHEY_SIMPLEX, 0.7, (0, 0, 255), 2)

    # 显示图像
    cv2.imshow('frame', frame)

    # 按下q键退出
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

# 释放摄像头并关闭窗口
cap.release()
cv2.destroyAllWindows()

使用此代码实现人数检测。首先通过cv2.VideoCapture()函数加载摄像头,然后使用cv2.CascadeClassifier()函数创建人脸检测器。在每个循环迭代中,读取摄像头数据并调整图像大小,然后将图像转换为灰度图像并使用人脸检测器检测人脸,最后绘制人脸矩形框并更新人数计数器。在图像上显示人数并使用cv2.imshow()函数显示图像,如果按下q键则退出循环。

需要安装OpenCV和imutils库,可以使用以下命令进行安装:

pip install opencv-python
pip install imutils

猜你喜欢

转载自blog.csdn.net/qq_46644680/article/details/129725164
今日推荐