一种轻量化目标检测算法研究

来源：计算机技术与发展杂志2020年第11期北京时间：

作者：余进;';史燕中;王春华;赵倩;吴蔚

摘要：基于深度卷积神经网络的目标检测算法对硬件的计算性能要求很高, 难以部署在一些嵌入式设备和移动终端中, 而当前的一些轻量化分类算法没有针对目标检测任务的特点进行网络结构设计。针对这一问题,借鉴深度可分离卷积的思路,通过引入多尺度的特征融合模块,设计了一个针对目标检测任务的轻量化特征提取网络 TinyNet,进而提高了轻量化特征提取网络对不同尺度目标的适应性。 结合当前性能较好的 YOLOv3 目标检测框架,用 TinyNet 取代 YOLOv3 的特征提取网络,并利用轻量化模块进一步优化 YOLOv3 的检测子网络,得到一个轻量化的目标检测模型 Tiny-YOLOv3+。实验结果表明,相比于使用其他轻量化特征提取网络设计的 YOLOv3,Tiny-YOLOv3+在检测精度有所提升的基础上,大大地降低了原模型的参数量,明显提升了检测速度,有效提高了轻量化检测模型的性能和效率。

关键词：深度学习;目标检测;轻量化;深度可分离卷积;YOLOv3