paper:ThunderNet: Towards Real-time Generic Object Detection
论文链接:https://arxiv.org/abs/1903.11752?context=cs.CV
开源代码:https://github.com/qixuxiang/Pytorch_Lightweight_Network 1
不得不说,ThunderNet 做的速度非常快。文章也有很多insight的东西。
官方的定义是这样的:第一个实现了在ARM平台上的实时检测器和最快的单线程速度。
个人觉得,输入分辨率和网络骨干能力的耦合度这一点是值得思考的。目前已经开源了,非常值得学习。
摘要:
移动平台上的实时通用对象检测是一项至关重要但具有挑战性的计算机视觉任务。然而,先前基于CNN的检测器遭受巨大的计算成本,这阻碍了它们在计算受限的场景中的实时推断。在本文中,我们研究了两级探测器在实时通用检测中的有效性,并提出了一种名为ThunderNet的轻量级两阶段目标检测器。在骨干部分,我们分析了先前轻量级骨架的缺点,并提出了一种用于物体检测的轻量级骨架。在检测部分,我们利用极其高效的RPN和检测头设计。为了生成更具辨别力的特征表示,我们设计了两个高效的架构块,上下文增强模块和空间注意模块。最后,我们研究