用python写一个口罩识别代码

好的，以下是一个简单的口罩识别代码示例，使用 TensorFlow 和 OpenCV 库：

import cv2
import tensorflow as tf

# 加载模型
model = tf.keras.models.load_model('model.h5')

# 加载分类标签
labels = ['no_mask', 'mask']

# 打开摄像头
cap = cv2.VideoCapture(0)

while True:
    # 读取摄像头中的图像
    ret, frame = cap.read()
    
    # 图像预处理
    img = cv2.resize(frame, (224, 224))
    img = img / 255.0
    img = img.reshape((1, 224, 224, 3))
    
    # 进行预测
    pred = model.predict(img)
    label = labels[pred.argmax()]
    
    # 显示结果
    cv2.putText(frame, label, (50, 50), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0), 2)
    cv2.imshow('mask detection', frame)
    
    # 按 'q' 键退出
    if cv2.waitKey(1) == ord('q'):
        break

# 释放摄像头和窗口
cap.release()
cv2.destroyAllWindows()

此代码使用了一个预先训练好的模型，该模型可以分类口罩和无口罩的图像。你可以根据你的需求来训练自己的模型，并使用该模型进行预测。