详解OpenVINO 模型库中的人脸检测模型

极市平台

共 1882字,需浏览 4分钟

 ·

2021-03-30 10:33

↑ 点击蓝字 关注极市平台

作者丨gloomyfish
来源丨OpenCV学堂
编辑丨极市平台

极市导读

 

本文介绍了OpenVINO的三种人脸检测模型:MobileNetv2、SqueezeNet和ResNet152。 >>加入极市CV技术交流群,走在计算机视觉的最前沿

OpenVINO的模型库中有多个人脸检测模型,这些模型分别支持不同场景与不同分辨率的人脸检测,同时检测精度与速度也不同。下面以OpenVINO2020 R04版本为例来逐一解释模型库中的人脸检测,列表如下:

从列表中可以看出骨干特征网络主要是MobileNetv2与SqueezeNet两种支持实时特征网络,ResNet152是高精度的特征网络,检测头分别支持SSD、FCOS、ATSS。

01 MobileNetv2版本的模型结构


上面三张图就是MobileNetv2的全部!看懂就好!

02 ResNet网络

残差网络模型在2015年提出,OpenCV从3.3版本开始支持,相关的论文如下:

https://arxiv.org/pdf/1512.03385.pdf

与VGG模型对比,模型结构:

03 SqueezeNet网络

该模型与MobileNet网络都是支持移动端/端侧可部署的模型,2016提出,相关论文如下:

https://arxiv.org/pdf/1602.07360.pdf

模型结构

检测模型

根据不同的检测头,组合生成不同的对象检测模型,这里三种常见的检测模型:

01 SSD检测

SSD论文中给出的是基于VGG-16作为backbone网络的,替换VGG为MobileNetV2,然后从第12个权重层开始提取特征到第14或者15个权重层,针对anchor预测与类别预测完成回归与分类损失计算,实现对象检测模型训练。

02 FCOS检测

跟YOLOv1相似是一个anchor-free的网络模型,多了一个中心输出来压制低质量的bounding box输出。

03 ATSS

ATSS的全称是自适应训练样本选择,

作者首先对比了RetinaNet与FCOS两种对象检测方法,发现对象检测在回归阶段的本质问题是如何选择正负样本,解决样本不平衡性问题,从而提出了一种新的正负样本选择定义方法-自适应训练样本选择(ATSS)。流程如下


该方法可以获得更好的对象检测精度与mAP,适用RetinaNet与FCOS等对象检测网络。实际实验数据对比:


代码演示与对比

上面的OpenVINO的人脸检测模型列表中,MobileNetv2 + SSD/FCOS适用于速度优先,不同分辨率的场景,ResNet152 + ATSS是OpenVINO模型库中的精度最高的人脸检测预训练模型。下面我们就通过代码分别测试FCOS与ATSS两种检测模型的检测效果比较,针对同一张测试人脸图像,图示如下:(此张非原图!原图太大,无法上传

统一调整阈值为0.25,推理时间与检测人脸总数的对比图如下:


最终分别得到测试图像如下:

FCOS:

ATSS:

极市 & 英特尔OpenVINO™联合(中级)认证地址:https://www.cvmart.net/list/OV2020
(官方权威认证证书+丰厚礼品+视频实操教程)


推荐阅读


OpenVINO部署Mask-RCNN实例分割网络

2021-03-14

实用教程详解:用OpenCV的DNN模块部署YOLOv5目标检测

2021-01-19

用OpenCV实现超轻量的NanoDet目标检测模型!

2021-02-18



# CV技术社群邀请函 #

△长按添加极市小助手
添加极市小助手微信(ID : cvmart2)

备注:姓名-学校/公司-研究方向-城市(如:小极-北大-目标检测-深圳)


即可申请加入极市目标检测/图像分割/工业检测/人脸/医学影像/3D/SLAM/自动驾驶/超分辨率/姿态估计/ReID/GAN/图像增强/OCR/视频理解等技术交流群


每月大咖直播分享、真实项目需求对接、求职内推、算法竞赛、干货资讯汇总、与 10000+来自港科大、北大、清华、中科院、CMU、腾讯、百度等名校名企视觉开发者互动交流~


△点击卡片关注极市平台,获取最新CV干货

觉得有用麻烦给个在看啦~  
浏览 23
点赞
评论
收藏
分享

手机扫一扫分享

举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

举报