RetiaFace 模型介绍
稳定调用及效果更好的API,详见视觉开放智能平台:人脸检测与五官定位。
人脸检测关键点模型RetiaFace
模型描述
RetiaFace为当前学术界和工业界精度较高的人脸检测和人脸关键点定位二合一的方法,被CVPR 2020 录取(论文地址, 代码地址)),该方法的主要贡献是:
- 引入关键点分支,可以在训练阶段引入关键点预测分支进行多任务学习,提供额外的互补特征,iferece去掉关键点分支即可,并不会引入额外的计算量。
RetiaFace 的结构如下:

模型效果

模型使用方式和使用范围
本模型可以检测输入图片中人脸以及对应关键点的位置。
使用方式
- 推理:输入图片,如存在人脸则返回人脸和人脸五点关键点位置,可检测多张人脸
目标场景
- 人脸相关的基础能力,可应用于人像美颜/互动娱乐/人脸比对等场景
模型局限性及可能偏差
- 小脸的检测效果一般
- 当前版本在pytho 3.7环境测试通过,其他环境下可用性待测试
预处理
测试时主要的预处理如下:
- Normalize:图像归一化,减均值除以标准差
- Pad:图像高宽补零至32的倍数
代码范例
from modelscope.pipelies import pipelie
from modelscope.utils.costat import Tasks
retia_face_detectio = pipelie(Tasks.face_detectio, 'damo/cv_reset50_face-detectio_retiaface')
img_path = 'https://modelscope.oss-c-beijig.aliyucs.com/test/images/retia_face_detectio.jpg'
result = retia_face_detectio(img_path)
prit(f'face detectio output: {result}.')
模型训练流程
- 在Wider Face数据集上使用SGD优化器,Warmup 5个epoch, 初始学习率为1e-2,batch size 为32,训练了80个epoch。
测试集
- WIDERFACE: 测试集已上传至ModelScope的DatasetHub, 详情请见WIDER_FACE。
数据评估及结果
模型在WiderFace的验证集上客观指标如下:
| Method |
Easy |
Medium |
Hard |
| RetiaFace |
94.8 |
93.8 |
89.6 |
人脸相关模型
以下是ModelScope上人脸相关模型:
来源说明
本模型及代码来自开源社区(地址),请遵守相关许可。
引用
如果你觉得这个该模型对有所帮助,请考虑引用下面的相关的论文:
@iproceedigs{deg2020retiaface,
title={Retiaface: Sigle-shot multi-level face localisatio i the wild},
author={Deg, Jiakag ad Guo, Jia ad Ververas, Evagelos ad Kotsia, Iree ad Zafeiriou, Stefaos},
booktitle={Proceedigs of the IEEE/CVF coferece o computer visio ad patter recogitio},
pages={5203--5212},
year={2020}
}
评论