0
点赞
收藏
分享

微信扫一扫

建立目标检测的模型,识别图中是否佩戴口罩

爱情锦囊 2022-05-04 阅读 114
java


今年一场席卷全球的新型冠状病毒给人们带来了沉重的生命财产的损失。有效防御这种传染病毒的方法就是积极佩戴口罩。我国对此也采取了严肃的措施,在公共场合要求人们必须佩戴口罩。在本次实验中,我们要建立一个目标检测的模型,可以识别图中的人是否佩戴了口罩。建立深度学习模型,检测出图中的人是否佩戴了口罩,并将其尽可能调整到最佳状态。学习经典的模型 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

举报

相关推荐

0 条评论