32 OPENVINO advanced course 使用DL-streamer进行表情识别

使用DL-streamer进行表情识别

初始化环境

#定义工作目录

export OV=/opt/intel/openvino_2021/

export WD=~/OV-300/07/

#初始化OpenVINO

source $OV/bin/setupvars.sh

开始实验

#进入实验目录

cd $WD

#播放原始视频示例：

show video1.mp4

运行识别示例

#运行人脸识别和分类示例：

bash face_detection_and_classification.sh video1.mp4

#获得output.mp4,获得可播放的标准MP4格式，使用：

ffmpeg -i output.mp4 output_1.mp4

#播放结果视频：

show output_1.mp4

挑战任务：增加情绪识别功能

#请你思考如何在年龄/性别识别的基础上，怎么情绪识别的功能：

修改 postproc_callbacks/age_gender_classification.py 脚本，我们往里添加的代码是:
if ‘prob_emotion’ == layer_name:

            emotions = ["neutral", "happy", "sad", "surprise", "anger"]

            tensor.set_label(emotions[numpy.argmax(data)])



            continue

修改 face_detection_and_classification.sh 脚本。需要和 age-gender-recognition模型一样，同理增加emotion-recognition的部分，所用 模型为emotions-recognition-retail-003.xml

注：答案可以在Solution中获取

cd Solution

#查看age_gender_classification.py/face_detection_and_classification.sh脚本的改动

bash face_detection_and_classification.sh video1.mp4

ffmpeg -i output.mp4 output_1.mp4

#播放结果视频：

show output_1.mp4