从0到1学习使用OAK——面部信息识别示例

编辑：OAK中国
来源：oakchina.cn
版权说明：转载请注明出处

前言

Hello，大家好，这里是OAK中国，我是助手君。

本期内容将为你介绍如何通过面部来判断人的情绪，年龄，性别，超级易上手，快来试试吧。👫👫👫

1. 分析项目

人的情绪，年龄，性别都可以通过面部表现出来，所有首先我们就要获取面部图像。然后就可以通过模型来识别情绪，年龄，性别。

2. 模型介绍

在OpenVINO的open_model_zoo(预训练模型库)中可以找到完成此项目的所需模型。
模型名称及其作用都列在下表中了：

模型名称	模型作用
face-detection-retail-0004	面部检测模型
emotions-recognition-retail-0003	情绪识别模型
age-gender-recognition-retail-0013	年龄，性别识别模型

模型剖析：
我们使用Netron工具查看模型结构。
a. face-detection-retail-0004
输入：

输出：

b. emotions-recognition-retail-0003
此模型可以识别五种情绪（“中性”，“快乐”，“悲伤”，“惊奇”，“愤怒”）。
输入：

输出：
模型输出形状为[1,5,1,1],表示五种情绪的概率。
c. age-gender-recognition-retail-0013
输入：

输出：
模型输出有两层：
“ age_conv3”：形状：[1、1、1、1]-估算年龄除以100。
“ prob”，形状：[1、2、1、1]-跨2个类型类的Softmax输出[female，male]

3. 转为 DepthAI项目

a. 首先下载并安装OpenVINO
b. 然后使用OpenVINO的模型下载脚本工具下载模型。
例如：

python ~/intel/openvino/deployment_tools/open_model_zoo/tools/downloader/downloader.py --name face-detection-retail-0004 --output_dir ~/open_model_zoo_downloads/

c. 此时得到的模型是OpenVINO IR，也就是.bin和.xml文件。我们还需要将IR转换为DepthAI所需要的blob文件。
将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安装。

4. 最终效果

5. 应用源代码
面部信息识别github链接

参考资料

https://github.com/OAKChina/depthai-examples/blob/master/facial_info_recognition/readme.rst
https://www.oakchina.cn/selection-guide/
https://docs.oakchina.cn/en/latest/

---

OAK中国
| OpenCV AI Kit在中国区的官方代理商和技术服务商
| 追踪AI技术和产品新动态

戳「+关注」获取最新资讯↗↗