请给我一个识别人脸是否戴口罩的python程序
时间: 2023-02-07 19:22:06 浏览: 75
这是一个简单的人脸检测和口罩检测的程序。它使用 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)
# 显示图像
cv2.imshow("Image", image)
cv2.waitKey