深度学习优质实践案例：用 Keras 实现人群中的口罩检测

2020-05-28 00:00:00 专区 订阅 模型 项目 权重

Keras 是一个由 Python 编写的开源人工神经网络库，可以作为 Tensorflow、Microsoft-CNTK 和 Theano 的高阶应用程序接口，进行深度学习模型的设计、调试评估、应用和可视化。

Gitee 上这位的开发者也是一名深度学习爱好者，为了做一次目标检测学习了 RCNN，FastRCNN，FasterRCNN 的理论，随后便诞生了这个项目，我们一起来看看他是怎么做的吧。

项目名称：keras 口罩检测

项目作者：唐振超

开源许可协议：GPL-3.0

项目地址：https://gitee.com/tang_zhen_chao/keraskouzhaojiance

介绍

使用 keras 搭建 fasterRCNN，在 VOC 格式的口罩数据集上训练，达到了检测人群中有无戴口罩的目的。

软件架构

*可前往项目主页查看

安装教程

  1. 需要numpy,matplotlib,scikit-learn,Pillow,tensorflow1.x,keras
  2. pip install package或者conda install package
  3. 如果要训练，需要使用VOC格式数据集，需要安装labelimg

使用说明

  1. ./model_data/logs下必须有模型权重文件，由于权重较大，所以未上传
  2. ./theory下是做这个东西的流程与方便理解代码的原理说明
  3. ./net下是fasterRCNN的模型构建
  4. run.py用于直接运行看结果，voctrain.py用于训练自己的VOC数据集

结果显示

点击链接前往项目主页看看项目的详细信息：gitee.com/tang_zhen_c…


相关文章