效果展示
一、 准备流程:
1. 在python环境中加载openvino
打开openvino安装目录如:
C:\Intel\openvino\python\python3.6
把目录下的openvino文件夹复制到
系统的python环境安装目录下如: C:\Python36\Lib\site-packages
2. 编译
C:\Intel\openvino\deployment_tools\inference_engine\samples 路径下执行:
build_samples_msvc2017.bat
执行完后在
C:\Users\kang\Documents\Intel\OpenVINO 目录
可以看到生成的
inference_engine_samples_build_2017 文件目录
在build目录中也可以找到cpu_extension:
cpu_extension = "C:\Users\kang\Documents\Intel\OpenVINO\inference_engine_samples_build_2017\intel64\Release\cpu_extension.dll"
3. 下载模型,记录路径
face-detection-adas-0001
emotions-recognition-retail-0003
model_xml = ""
model_bin = ""
二、 参数介绍:
1. emotions提取
基于MobileNet v1版本
· 输入格式:[1x3x384x672]= BCHW
· 输出格式:[1, 1, N, 7] = [image_id, label, conf, x_min, y_min, x_max, y_max]
表情识别网络 – 输入-[1x3x64x64]=BCHW
· 输出格式- [1, 5, 1, 1]
· 检测五种表情 (‘neutral’, ‘happy’, ‘sad’, ‘surprise’, ‘anger’)
2. python版本的api介绍
同步调用,执行输入
landmark_res = exec_emotions_net.infer(inputs={input_blob: [face_roi]})
3. 获取输出
landmark_res = landmark_res[‘prob_emotion’]
landmark_res = np.reshape(landmark_res, (5))
landmark_res = labels[np.argmax(landmark_res)]
三、 附录代码:
import sys
import cv2
import numpy as np
import time
import logging as log
from openvino.inference_engine import IENetwork, IEPlugin
plugin_dir = "C:/Intel/openvino/deployment_tools/inference_engine/bin/intel64/Release"
cpu_extension = "C:/Users/kang/Documents/Intel/OpenVINO/inference_eng