Samples
with mask
no mask
Overview
这个口罩数据集主要用来进行分类(有无口罩)以及检测(口罩位置),其中有口罩的有690张jpg图像,无口罩的有686张jpg图像。
整个数据集的制作也是非常有趣,忍不住分享一波。
首先自然是无口罩图像的采集,比如
然后通过人脸检测算法获取人脸边界框
接着就可以提取人脸的ROI区域了,也就是
到了这一步就可以利用dlib库检测面部特征了,目的是方便知道口罩在人脸上的放置位置
现在只需要找一个口罩图片,比如
最后只要根据鼻子和下巴的位置,适当调整口罩的大小和方向就可以给人物带上口罩了
乍一看是不是还挺逼真~
最后的最后对所有图像重复该操作就可以实现文章开头的with mask数据集了
总结一下:
1、拍摄并采集大量的人脸图像(爬虫啥的自然也可以哦)
2、基于tensorflow创建一个py脚本给人脸添加口罩,创建一个artificial mask数据集
扫码关注后,后台回复 mask 即可获得