模型仅1MB,更轻量的人脸检测模型开源,效果不弱于主流算法

作者: 乾明 2019-10-14 15:13:24

模型仅1MB,更轻量的人脸检测模型开源,效果不弱于主流算法

本文经AI新媒体量子位(公众号ID:QbitAI)授权转载,转载请联系出处。

AI模型越来越小,需要的算力也也来越弱,但精度依旧有保障。

最新代表,是一个刚在GitHub上开源的中文项目:一款超轻量级通用人脸检测模型。

项目贡献者介绍,这一模型大小文件仅1MB,320x240输入下计算量仅90MFlops。

当然,效果也不弱于当前业界主流的开源人脸检测算法,甚至有所超越。

超轻量、通用

这一模型的贡献者为linzai,他介绍称,这是针对边缘计算设备或低算力设备(如用ARM推理)设计的一款实时超轻量级通用人脸检测模型:

默认FP32精度下(.pth)文件大小为 1.1MB,推理框架int8量化后大小为 300KB 左右。

模型仅1MB,更轻量的人脸检测模型开源,效果不弱于主流算法

目标就是在低算力设备中用ARM进行实时的通用场景的人脸检测推理。同时,这也适用于移动端环境(Android & IOS)、PC环境(CPU & GPU )等等。

GitHub项目页面介绍称,模型已经在Ubuntu16.04、Ubuntu18.04、Windows 10;Python3.6;Pytorch1.2;CUDA10.0 + CUDNN7.6等环境进行过测试,都能够保证正常运行。

模型设计,一共有两个版本,分别是:1)version-slim,主干精简速度略快,2)version-RFB,加入了修改后的RFB模块,精度更高。

而且还提供了320x240、640x480不同输入分辨率下使用widerface训练的预训练模型,能够更好的工作于不同的场景。

linzai介绍称,整个项目中没有特殊算子,支持onnx导出,从而便于移植推理。

效果不弱于当前主流开源算法

模型仅1MB,更轻量的人脸检测模型开源,效果不弱于主流算法

如此模型,效果/精度到底如何?

linzai也在GitHub项目页面放出了这一模型在精度、速度、场景测试、大小方面的测试。

参赛选手有两个,一个是Libfacedetection,OpenCV中文站站长于仕琪开源的人脸检测算法。

另一个是Retinaface-Mobilenet-0.25 (Mxnet)——当前业界最先进的开源人脸检测算法之一。

在Widerface数据集上的测试结果如下:

模型仅1MB,更轻量的人脸检测模型开源,效果不弱于主流算法

version-slim/version-RFB基本上都能够实现最优结果。

在树莓派4B MNN推理测试耗时上的测试结果为:

模型仅1MB,更轻量的人脸检测模型开源,效果不弱于主流算法

对比之下,version-slim版本的速度也不在下风。

在地铁站、万达广场、办公室等场景进行测试,结果上version-RFB也比较有优势。

模型仅1MB,更轻量的人脸检测模型开源,效果不弱于主流算法

更重要的是,新开源模型的轻量性:

模型仅1MB,更轻量的人脸检测模型开源,效果不弱于主流算法

传送门

在这个项目的GitHub页面,linzai也分享了如何生成VOC格式训练数据集以及训练流程,以及如何更好的使用这一模型的方法。

如果你有兴趣,不妨看一看~

AI 数据 人工智能
上一篇:人工智能遇冷,自动驾驶受阻? 下一篇:全球十大AI训练芯片大盘点
评论
取消
暂无评论,快去成为第一个评论的人吧

更多资讯推荐

人工智能和机器人“统治”医疗业,你放心吗?

人类社会发展至今,智能已成为新的生产要素。近年来,人工智能行业发展迎来了爆发期,整个行业开始重新认识技术创新的价值。

读芯术 ·  15h前
盘点2019人工智能:产业发展已驶入快车道

2019年,中国人工智能市场“狂欢”—— 据IDC统计,2019年全球人工智能市场规模达到375亿美元,中国人工智能市场预计占全球的12%,成为了全球第二大人工智能单一市场,且市场规模保持高速增长。

蒋雅丽 刁兴玲 ·  19h前
2019 AI Index 报告出炉:AI 领域取得的进展很多,但结果忧喜参半

斯坦福「以人为本」人工智能研究院与 OpenAI 合作完成的2019 AI Index 年度报告发布。

丛末 ·  1天前
腾讯云重磅发布七大新品,AI应用走向精细化时代

12月11日,腾讯云在北京举行大数据AI新品发布会。会上,腾讯云带来了在大数据与AI领域的最新研究成果,包括AI换脸甄别技术AntiFakes、腾讯星图以及企业画像平台等七大重磅新品,并对AI、大数据产品进行全线升级,致力于为用户带来更精细化的应用场景、更强大的技术能力以及更低的应用成本,全面降低企业AI技术应用门槛。

佚名 ·  1天前
谭茗洲:技术创新者—少有人走的路│LeaTech全球CTO领导力峰会回顾

日前,由51CTO旗下CTO训练营品牌打造的LeaTech全球CTO领导力峰会隆重召开。峰会邀请远望智库合伙人谭茗洲带来主题分享。他主要围绕两个方面展开论述:如何从技术视角看待未来产业发展的趋势,以及作为技术创业者如何把自己的优势在技术驱动的年代里转化成自身的势能。

张洁 ·  1天前
AI公开课 | 我猜你对数据标注还不了解吧?

当我们打开手机,启用美颜APP里各种各样的AI滤镜时,我们看到的是搞笑的头像和可爱的表情,而APP“看到”的是我们脸上数百个点。

班主任 ·  1天前
人工智能市场的需求与应用

“人工智能”连续三年进入政府工作报告,2019提出深化大数据、人工智能等研发应用拓展“智能+”,为制造业转型升级赋能。

泰一数据 ·  1天前
人工智能将成为电网和泛在电力物联网建设发展的重要方向

当前,在移动互联网、大数据、超级计算等新理论新技术和经济社会发展需求的共同驱动下,人工智能正迅速发展,渗透进各个行业,深刻改变着人们的生产生活方式。

AET电子技术应用 ·  2天前
Copyright©2005-2019 51CTO.COM 版权所有 未经许可 请勿转载