今年一场席卷全球的新型冠状病毒给人们带来了沉重的生命财产的损失。有效防御这种传染病毒的方法就是积极佩戴口罩。我国对此也采取了严肃的措施,在公共场合要求人们必须佩戴口罩。在本次实验中,我们要建立一个目标检测的模型,可以识别图中的人是否佩戴了口罩。建立深度学习模型,检测出图中的人是否佩戴了口罩,并将其尽可能调整到最佳状态。学习经典的模型 MTCNN 和 MobileNet 的结构。学习训练时的方法。实验使用重要python包:
import cv2 as cv
import numpy as np
import matplotlib.pyplot as plt
from tensorflow.keras.callbacks import ModelCheckpoint, ReduceLROnPlateau, EarlyStopping
由于担心平台GPU时长不够用,所以在自己电脑上搭建了配套实验环境,由于电脑显卡CUDA版本较老,所以最终本地配置如下:
Python: 3.8
Tensorflow-GPU: 2.3.0
Keras: 2.7.0
代码托管在了WRITE-BUG上面了:https://www.writebug.com/git/shuren/MaskRecognition