利用retinaface人脸检测和sphereface人脸识别训练好的模型在香橙派上进行联合推理
·
一:对单个图片进行联合推理 (训练有归一化则推理时也加上相同的归一化操作)
流程如下:1.对原始图片修改其尺寸并转为blob格式[1,3,640,640],并生成预测框;
2.将图片利用retinaface进行推理session.infer(),返回位置偏移量和类别置信度;
3.将位置偏移量和预测框利用utils工具进行解码,生成真实位置并映射到原始图片尺寸的位置信息
4.根据人脸置信度去除置信度低的位置,信息;并经过NMS非极大值抑制去除重叠率(IOU)高的信息
5.根据保留下来的位置框对原始图片进行裁剪裁剪出人脸(转换为BGR格式),并根据其对应的五官关键点进行仿射变换,相似变换生成标准脸
6.将标准脸修改其尺寸并转换为blob格式[1,3,112,96],利用sphereface进行推理,提取出特征向量
7.对提取出来的特征向量进行L2归一化,并与对比数据库中的所有特征向量都进行余旋相似度计算,取出相似度最好的那个并返回其对应的人名
8.如果相似度大于阈值,则是同一人,否则未备案
9.利用opencv在原图上绘制出位置框和添加人名
DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。
更多推荐

所有评论(0)