【每天进步一点点】基于ModelArts,跑通口罩检测案例(Python版本)——数据集部分
前些天,看了毛老师的直播【玩转Atlas 200 DK系列1:口罩识别应用案例】,回放链接: http://live.vhall.com/618554206, 老师不仅讲解了具体的代码和实现流程,还对Atlas系列产品做了介绍,感兴趣的同学可以看看哈。
看了直播,特别想自己跑一下,但我的是老版本,尚未升级到最新的版本,无法部署。纸上得来终觉浅,绝知此事要躬行,既然不行,那我我就自己训练一下,部署到Atlas 200 DK上试试吧,正好也检验一下工程部署的能力,体验一下ModelArts与Atlas 200 DK的端云协同。说干就干吧。在这里也分享一下,
1. 从数据集说起
数据可是一切的根本,从一定程度来说,数据决定了算法效果的上限。这里从数据集开始,完全基于ModelArts,使用的是一个公开数据集,来源链接为https://gitee.com/vincent_wd/FaceMaskDetection?_from=gitee_search。而把毛老师介绍中的500张图片做了测试集。具体在ModelArts上的操作可参考https://bbs.huaweicloud.com/blogs/163392
首先,上传数据集到OBS桶;
然后,在ModelArts创建数据集,由7958张图片组成,比老师用的500张图片多得多,理论上应该可以取得更好的效果,而且这里做了简化,简单粗暴的把戴口罩和没戴口罩的图像作为输入处理,不涉及人脸和口罩的重叠程度问题,来看看:
还不错吧。使用ModelArts可以轻松创建并管理数据集,而且还能对数据进行分析,如图所示,这些特征都可以看到:
在这里,我们具体来看看单张图片中框的个数:
怎么样,是不是特别方便,其实,还有很多方便的功能,期待大家去发掘吧。基于ModelArts开发Atlas 200 DK应用也很方便,不过这里重点是Atlas 200 DK,至于ModelArts,大家可以去看看。
未完待续......
最终效果(绿色表示检测到戴了口罩,红色表示检测到未戴口罩)
- 点赞
- 收藏
- 关注作者
评论(0)