Keras 是一个由 Python 编写的开源人工神经网络库,可以作为 Tensorflow、Microsoft-CNTK 和 Theano 的高阶应用程序接口,进行深度学习模型的设计、调试评估、应用和可视化。
Gitee 上这位优秀的开发者也是一名深度学习爱好者,为了做一次目标检测学习了 RCNN,FastRCNN,FasterRCNN 的理论,随后便诞生了这个项目,我们一起来看看他是怎么做的吧。
项目名称:keras 口罩检测
项目作者:唐振超
开源许可协议:GPL-3.0
项目地址:https://gitee.com/tang_zhen_chao/keraskouzhaojiance
介绍
使用 keras 搭建 fasterRCNN,在 VOC 格式的口罩数据集上训练,达到了检测人群中有无戴口罩的目的。
软件架构
*可前往项目主页查看
安装教程
- 需要numpy,matplotlib,scikit-learn,Pillow,tensorflow1.x,keras
- pip install package或者conda install package
- 如果要训练,需要使用VOC格式数据集,需要安装labelimg
使用说明
- ./model_data/logs下必须有模型权重文件,由于权重较大,所以未上传
- ./theory下是做这个东西的流程与方便理解代码的原理说明
- ./net下是fasterRCNN的模型构建
- run.py用于直接运行看结果,voctrain.py用于训练自己的VOC数据集
结果显示
点击链接前往项目主页看看项目的详细信息:https://gitee.com/tang_zhen_chao/keraskouzhaojiance
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:深度学习优质实践案例:用 Keras 实现人群中的口罩检测 - Python技术站