资讯专栏INFORMATION COLUMN

国防科大、旷视提出首个在ARM上实时运行的通用目标检测算法

CoderDock / 1636人阅读

摘要:算法思想作者是在经典的两阶段目标检测算法的基础上做的改进。使用轻量级面向目标检测任务设计的骨干网。二改进特征表示的鉴别性。使用骨干网的精度超过,而计算量仅为其。

今天跟大家分享一篇前天新出的论文《ThunderNet: Towards Real-time Generic Object Detection》,来自国防科大与旷视的研究团队(孙剑老师在列)提出了较早的能够在移动端ARM芯片实时运行的两阶段通用目标检测算法ThunderNet(寓意像Thunder雷一样快^_^),并称该算法后续将开源!

以下是作者信息:

论文地址:

https://arxiv.org/pdf/1903.11752v1.pdf

一作为来自国防科大的Zheng Qin和旷视的Zeming Li。

提出问题

目标检测的算法有很多,但要在移动平台,比如手机上的ARM CPU实时还比较难,主流的state-of-the-art目标检测算法无法实时,而一些号称可以跑起来的算法距离state-of-the-art的精度差距太大。

作者希望设计速度快而精度高的算法,满足移动端的部署需求。

下图是该文提出的ThunderNet算法的三个版本与目前轻量级检测网络在COCO test-dev数据集上的精度和计算量的比较图,可见ThunderNet在精度达到或超过之前轻量级检测网络的同时,降低了计算量。

以上均为在CPU上单线程运行结果,MobileNet-SSD运行在骁龙820,MobileNet/MobileNetV2-SSDLite运行在骁龙810,Pelee运行在Intel i7-6700K (4.0 GHz),ThunderNet运行在骁龙845。

算法思想

作者是在经典的两阶段目标检测算法的基础上做的改进。

网络结构如下:

作者分两个方向做了5点改进:

一、改进速度。

1. 使用轻量级、面向目标检测任务设计的骨干网Snet。

网络参数如下:

2. 压缩RPN网络部分,降低候选目标区域生成的时间。

3. R-CNN子网络使用1024维的全连接层,提高速度。

二、改进特征表示的鉴别性。

1. 设计了上下文增强模块CEM结构,它可以有效结合三个尺度的特征图,编码更多的上下文信息,增大感受野,生成更具鉴别性的特征。

2. 设计了空间注意力模块SAM结构,它可以利用RPN中学习到的信息,从上下文增强模块提精特征图的特征分布。

实验结果

作者在VOC 2007 test数据集上的实验结果如下,与现有其他算法相比,在保持高精度同时,计算量也处于较低水平。

下图是在COCO test-dev数据集上与其他算法的比较,同样达到了计算量小精度高的目标。

以下是一些检测结果示例:

在运行速度方面,该算法能到达到与MobileNet-SSD相当的精度,此时在ARM上运行24.1fps,x86上运行47.3fps。使用SNet146骨干网的ThunderNet精度超过YOLOv2,而计算量仅为其1/37。

期待作者早日放出代码!

声明:本文版权归原作者所有,文章收集于网络,为传播信息而发,如有侵权,请联系小编及时处理,谢谢!欢迎加入本站公开兴趣群

商业智能与数据分析群

兴趣范围包括各种让数据产生价值的办法,实际应用案例分享与讨论,分析工具,ETL工具,数据仓库,数据挖掘工具,报表系统等全方位知识

QQ群:81035754

文章版权归作者所有,未经允许请勿转载,若此文章存在违规行为,您可以联系管理员删除。

转载请注明本文地址:https://www.ucloud.cn/yun/4871.html

相关文章

  • AI Challenger开赛,千万量级数据开放,AI高手将上演巅峰对决

    摘要:月日,各项竞赛的排名将决定最终的成绩排名。选手通过训练模型,对虚拟股票走势进行预测。冠军将获得万元人民币的奖励。 showImg(https://segmentfault.com/img/bVUzA7?w=477&h=317); 2017年9月4日,AI challenger全球AI挑战赛正式开赛,来自世界各地的AI高手,将展开为期三个多月的比拼,获胜团队将分享总额超过200万人民币的...

    Ali_ 评论0 收藏0
  • 首次披露!阿里线下智能方案进化史

    摘要:前言阿里巴巴机器智能实验室线下智能团队从年底开始涉及线下智能领域,从算法工程产品化业务落地多个方面入手,与合作伙伴们一起取得了一些小小的成绩。目前,该套工具作为推荐的量化工具广泛应用在阿里集团内多个线下业务场景中。 showImg(https://segmentfault.com/img/remote/1460000019246850); 阿里妹导读:AI 技术已经从互联网走向零售、...

    keelii 评论0 收藏0
  • 当区块链遇见 RISC-V

    摘要:在区块链上,虚拟机就是智能合约的运行环境,是一个可以完全对外隔离的完整计算机体系。区块链通过虚拟机来调用和执行智能合约,并要求所有节点都达成一致。当区块链遇见在很多科技领域都得到了运用,目前,也开始在区块链领域逐渐的得以发展。 showImg(https://segmentfault.com/img/bVbsfi2?w=2779&h=1179); 区块链的出现使得智能合约得到了更好的实...

    Tychio 评论0 收藏0

发表评论

0条评论

最新活动
阅读需要支付1元查看
<