基于深度学习的口罩佩戴检测
首先用现有的人脸检测算法,直接对图像进行人脸检测,训练好的人脸分类权值文件last_one.h5已放到log文件夹下,然后将检测的每一个人脸,单独切割出来,进行是否戴口罩的二分类。先对图像中的人脸进行标注,包括了戴口罩的,和不带口罩的两个label的目标。佩戴口罩的人脸图片命名标记为mask_1,没有佩戴口罩的人脸图片为nomask_1,然后在数据集准备好之后可以利用train.py文件训练 ,效果与详解可以看bilibili视频:https://www.bilibili.com/video/BV1dZ4y1F7Po/
文件列表
基于深度学习的口罩佩戴检测
(预估有个1950文件)
onet.h5
1.53MB
pnet.h5
56KB
rnet.h5
428KB
val_400.jpg
1.82MB
val_186.jpg
4.38MB
val_190.jpg
4.81MB
.gitignore
233B
mask-recognize-master.iml
348B
last_one.h5
12.52MB
ep002-loss0.401-val_loss0.519.h5
12.52MB
暂无评论