标签：DepthAI 检测 模型 mask 口罩 sbd onnx
这篇文章将介绍如何在DepthAI上实现口罩检测。
1.查找开源项目
通过搜索引擎我们能够找到一个叫做sbd_mask的开源项目，它是思百达开源的一款轻量级实时口罩检测项目。
2.分析项目
通过分析sbd_mask项目，我们可以得到以下信息:
- 使用的模型及其作用
| 模型 | 作用 |
|---|---|
| centerface.onnx | 用于检测人脸 |
| sbd_mask.onnx | 用于检测是否佩戴口罩 |
- 应用工作流程:
| a. 从相机获取图片 |
|---|
| b. 运行centerface检测人脸 |
| c. 将centerface检测到的人脸送入sbd_mask检测是否佩戴口罩 |
| d. 输出结果 |
3.转为 DepthAI项目
1）替换centerface模型
此模型的作用是检测人脸，我们使用openvino open_model_zoo中的face-detection-retail-0004模型替换此模型，因此无需进行转换。
2）转换sbd_mask.onnx模型
因为DepthAI需要特定的模型格式，所以需要对sbd_mask.onnx模型进行转换。
首先转换为OpenVINO IR:
python mo_onnx.py --input_model sbd_mask.onnx --model_name sbd_mask --scale 255 --reverse_input_channels
再将OpenVINO IR编译为blob文件:
export MYRIAD_COMPILE=$(find /opt/intel/ -iname myriad_compile)
$MYRIAD_COMPILE -m sbd_mask.xml -ip U8 -VPU_MYRIAD_PLATFORM VPU_MYRIAD_2480 -VPU_NUMBER_OF_SHAVES 4 -VPU_NUMBER_OF_CMX_SLICES 4
注：使用模型转换工具需要安装openvino,安装时要注意DepthAI对OpenVINO版本有要求。 详情请参照openvino安装。
3）剖析模型
利用Netron可以查看模型结构。
模型输入：
模型输出：
由上图可知此模型输出只有“349”一层包含有两个数据。
输入输出总览:
4）编写解析代码
模型输出的解析代码可以参考原始的应用。
4.应用源代码
国内在哪里能买到DepthAI?
目前国内唯一的购买渠道是派驰的淘宝店:
OAK-D -派驰电子 淘宝
OAK-1 -派驰电子 淘宝
标签：DepthAI,检测,模型,mask,口罩,sbd,onnx 来源： https://blog.csdn.net/m0_54226098/article/details/113375695
本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享； 2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关； 3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关； 4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除； 5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。