请给我一个识别人脸是否戴口罩的python程序-蒲公英云

请给我一个识别人脸是否戴口罩的python程序

灰太狼 2024-03-25 14:56 163阅读 0赞

这是一个简单的人脸检测和口罩检测的程序。它使用 OpenCV 库来进行人脸检测，并使用一个预先训练的神经网络模型来识别人脸是否戴着口罩。

首先，你需要安装 OpenCV 和一个深度学习库，例如 TensorFlow 或 Keras。

```python import cv2 import tensorflow as tf

加载人脸检测器

face_cascade = cv2.CascadeClassifier(“haarcascade_frontalface_default.xml”)

加载口罩检测模型

model = tf.keras.models.load_model(“mask_detector.h5”)

读取图像并转换为灰度图

image = cv2.imread(“image.jpg”) gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

检测人脸

faces = face_cascade.detectMultiScale(gray, 1.3, 5)

循环每个人脸

for (x, y, w, h) in faces: # 截取人脸区域 face_region = image[y:y+h, x:x+w]

# 缩放图像并调整大小
face_region = cv2.resize(face_region, (150, 150))
face_region = face_region / 255.0
# 预测是否戴着口罩
prediction = model.predict(face_region[None, :, :, :])
# 如果预测值大于0.5，则表示戴着口罩
if prediction[0][0] > 0.5:
    cv2.putText(image, "Mask", (x, y-10), cv2.FONT_HERSHEY_SIMPLEX, 0.7, (0, 255, 0), 2)
else:
    cv2.putText(image, "No Mask", (x, y-10), cv2.FONT_HERSHEY_SIMPLEX, 0.7, (0, 0, 255), 2)