运行此爬虫需要配以一个name.txt,在其中寫下所需要爬取的信息即可,rename.py是对获取的数据进行命名整理的代码如下:
这里笔者采用的数据清洗就比较的笨了,就是人为一个一个嘚筛选整理了一个晚上,请原谅笔者的愚蠢
标注数据集同数据的清洗而言就比较简单了,比较的机械化这里笔者使用labelImg这个标注工具進行标注。这里做的是简单的二分类任务即:Mask和unMasked.标注比数据清洗要快一点。
准备工作的话就是使用一个版本的大体框架这里使用的是yolov3,下载后可以重新命名,当然也可不命名这个随性就可以了。
申明一下最后会将数据集提供给大家
值得注意的是这里是对人是否佩戴口罩进行识别的,即分为两类佩戴口罩和为佩戴口罩所以这里需要将classes设置为2.
这里需要将cfg下的yolov3-tiny.cfg文件进行修改,修改内容如下:
完成了上述操作后即可进行数据集的训练操作了。
在完成了训练后即可对训练的模型进行测试了,看一下这个其对佩戴口罩和未佩戴口罩者的預测效果
在操作的过程中难免会出现一些小毛病这里呢,笔者已经为这个项目总结了几个容易出现的问题请参考笔者的另一篇博客:
這里笔者将本项目的源代码和数据集都已经上传到github上,欢迎大家扩展为保护我们的工作者减少工作压力。
由于笔者知识能力有限在描述上可能存在不准确的地方,还请谅解
版权声明:文章内容来源于网络,版权归原作者所有,如有侵权请点击这里与我们联系,我们将及时删除。
点击添加站长微信