口罩检测.md

口罩检测器

1. 接口简介

口罩检测器根据输入的图像数据、人脸位置，返回是否佩戴口罩的检测结果。

2. 类型说明

2.1 struct SeetaImageData

名称	类型	说明
data	unit8_t*	图像数据
width	int32_t	图像的宽度
height	int32_t	图像的高度
channels	int32_t	图像的通道数
说明：存储彩色（三通道）或灰度（单通道）图像，像素连续存储，行优先，采用 BGR888 格式存放彩色图像，单字节灰度值存放灰度图像。

2.2 struct SeetaRect

名称	类型	说明
x	int32_t	人脸区域左上角横坐标
y	int32_t	人脸区域左上角纵坐标
width	int32_t	人脸区域宽度
height	int32_t	人脸区域高度

3 class MaskDetector

口罩检测器。

3.1 Enum SeetaDevice

模型运行的计算设备。

名称	说明
SEETA_DEVICE_AUTO	自动检测，会优先使用 GPU
SEETA_DEVICE_CPU	使用CPU计算
SEETA_DEVICE_GPU	使用GPU计算

3.2 struct SeetaModelSetting

口罩检测器需要传入的结构体参数。

参数	类型	缺省值	说明
model	const char**		检测器模型
id	int		GPU id
device	SeetaDevice	AUTO	计算设备(CPU 或者 GPU)

3.3 构造函数

MaskDetector

构造检测器，需要在构造的时候传入检测器结构参数。

参数	类型	缺省值	说明
setting	const SeetaModelSetting&		识别器接口参数

3.4 成员函数

detect

输入图像数据、人脸位置，返回是否佩戴口罩的检测结果。

参数	类型	缺省值	说明
image	const SeetaImageData&		原始图像数据
face	const SeetaRect&		人脸位置
score	float*	nullptr	戴口罩的置信度
返回值	bool		true为佩戴了口罩