zhengmingzhang / mask-detection

使用yolov3 pytorch版训练出一个口罩检测模型,没带口罩的话会进行语音提醒
21 stars 7 forks source link

mask-detetcion

项目说明

针对当前疫情情况,各大科技公司开始使用科技手段助力防疫。本项目通过yolov3对人脸口罩数据集进行训练,通过该模型可以识别一个人是否带了口罩,如果识别出此人没有佩戴口罩,则语音播报“请佩戴口罩”,如果此人是佩戴了口罩的,那么系统不会有反应。

数据集

用来训练的数据集来源于B站UP主:HamlinZheng,非常感谢该UP主的无私奉献~

如何使用该项目

链接:https://pan.baidu.com/s/11z6hmBitSHG4TjilDNFJfQ 提取码:2zvl

将best.pt放入weights文件夹中,创建一个data文件夹将mask.data和mask.name放进去,配置好项目所需的环境之后,在命令行执行

python detect.py --data-cfg data/mask.data --cfg cfg/yolov3-tiny-mask.cfg --weights weights/best.pt 便可看到结果。

语音播报的声音是我自己录的,不咋好听,如果想录自己的语音提醒可以直接运行record.py,录下自己的语音

模型我使用的是yolov3-tiny,模型训练结果实际上不算特别好,计划这两天对数据进行增强,优化一下模型。

如何训练自己的数据集

想要使用自己的数据集进行重新训练,需要对数据集的格式改成VOC数据集的形式,这个在网上有非常多的教程,我训练的时候参考的是 https://blog.csdn.net/qq_21578849/article/details/84980298 大家感兴趣也可以自己训练一下