71 lines
1.9 KiB
Markdown
71 lines
1.9 KiB
Markdown
# 口罩检测器
|
|
|
|
## **1. 接口简介** <br>
|
|
|
|
口罩检测器根据输入的图像数据、人脸位置，返回是否佩戴口罩的检测结果。<br>
|
|
|
|
## **2. 类型说明**<br>
|
|
|
|
### **2.1 struct SeetaImageData**<br>
|
|
|
|
|名称 | 类型 | 说明|
|
|
|---|---|---|
|
|
|data|unit8_t* |图像数据|
|
|
|width | int32_t | 图像的宽度|
|
|
|height | int32_t | 图像的高度|
|
|
|channels | int32_t | 图像的通道数|
|
|
说明：存储彩色（三通道）或灰度（单通道）图像，像素连续存储，行优先，采用 BGR888 格式存放彩色图像，单字节灰度值存放灰度图像。
|
|
|
|
### **2.2 struct SeetaRect**<br>
|
|
|
|
|名称 | 类型 | 说明|
|
|
|---|---|---|
|
|
|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为佩戴了口罩|
|