口罩检测的算法设计

两阶段检测器:先检测人脸,然后将人脸进行分类,戴口罩与不戴口罩

首先用现有的人脸检测算法,直接对图像进行人脸检测,然后将检测的每一个人脸,单独切割出来,进行是否戴口罩的二分类。

所以只需要训练一个 CNN 网络,包含两类图像,mask 和 without_maks。前端接一个 face detector,就能应对大多数情况了。

一阶段检测器:直接训练口罩检测器,训练样本为人脸的标注信息,戴口罩的,不带口罩的

先对图像中的人脸进行标注,包括了戴口罩的,和不带口罩的两个 label 的目标。然后进行标注。直接对人脸检测模型进行 transfer learning,微调一下模型,就能检测了。

直接对戴口罩和不带口罩的人脸进行标注就可以了。