26

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

 4 years ago
source link: https://www.tuicool.com/articles/N3qEFv6
Go to the source link to view the article. You can view the picture content, updated content and better typesetting reading experience. If the link is broken, please click the button below to view the snapshot at that time.

乾明 编辑整理

量子位 报道 | 公众号 QbitAI

U3y6buM.jpg!web

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

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

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

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

超轻量、通用

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

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

f2QbUrQ.jpg!web

目标就是在低算力设备中 用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导出,从而便于移植推理。

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

IJFfya7.jpg!web

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

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

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

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

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

fUbEV3i.jpg!web

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

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

73UfmyI.png!web

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

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

b6FNven.png!web

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

EvmYNjj.png!web

传送门

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

如果你有兴趣,请收好传送门:

https://github.com/Linzaer/Ultra-Light-Fast-Generic-Face-Detector-1MB

作者系网易新闻·网易号“各有态度”签约作者

— 完 —

大会启幕!预见智能科技新未来

量子位MEET 2020智能未来大会 启幕,将携手优秀AI企业、杰出科研人员呈现一场高质量行业盛会!详情可点击图片:

INJBfqf.jpg!web

榜单征集!三大奖项,锁定AI Top玩家

2019中国人工智能年度评选 启幕,将评选领航企业、商业突破人物、最具创新力产品3大奖项,并于MEET 2020大会揭榜,欢迎优秀的AI公司扫码报名!

VJzI7ba.jpg!web

iUzyMrz.jpg!web

量子位  QbitAI · 头条号签约作者

վ'ᴗ' ի 追踪AI技术和产品新动态

喜欢就点「好看」吧 !


About Joyk


Aggregate valuable and interesting links.
Joyk means Joy of geeK