红外与激光工程, 2018, 47 (7): 0703005, 网络出版: 2018-08-30   

基于改进SSD的轻量化小目标检测算法

A lightweight small object detection algorithm based on improved SSD
作者单位
1 沈阳工业大学 软件学院, 辽宁 沈阳 110870
2 中国科学院沈阳自动化研究所, 辽宁 沈阳 110016
摘要
为提高SSD目标检测算法的小目标检测能力, 提出在SSD算法中引入转置卷积结构, 采用转置卷积将低分辨率高语义信息特征图与高分辨率低语义信息特征图相融合, 增加低层特征提取能力, 提高SSD算法的平均精准度。同时针对SSD算法存在模型过大, 运行内存占用量过高, 无法在嵌入式ARM设备上运行的问题, 以DenseNet为基础, 结合深度可分离卷积,逐点分组卷积与通道重排提出轻量化特征提取最小单元, 将SSD算法特征提取部分替换为轻量化特征提取最小单元的组合后, 可在嵌入式ARM设备上运行。在PASCAL VOC数据集和KITTI自动驾驶数据集上进行对比实验, 结果表明改进后的网络结构在平均精准度上得到明显提升, 模型参数数量得到有效降低。
Abstract
In order to improve the small object detection ability of SSD object detection algorithm, the transposed convolution structure in SSD algorithm was proposed, the low resolution high semantic information feature map was integrated with high resolution low semantic information feature map using transposed convolution, which increased the ability of low level feature extraction and improved the average accuracy of SSD algorithm. At the same time for the problem that SSD algorithm model being large, running memory consumption high, without running on the embedded equipment ARM, a lightweight feature extraction minimum unit was proposed based on DenseNet, combining depthwise separable convolutions, pointwise group convolution and channel shuffle, running on the embedded equipment ARM cloud be realized. The comparative experiments on PASCAL VOC data set and KITTI autopilot data set show that the mean average is significantly improved by improved network structure, and the number of model parameters is effectively reduced.

吴天舒, 张志佳, 刘云鹏, 裴文慧, 陈红叶. 基于改进SSD的轻量化小目标检测算法[J]. 红外与激光工程, 2018, 47(7): 0703005. Wu Tianshu, Zhang Zhijia, Liu Yunpeng, Pei Wenhui, Chen Hongye. A lightweight small object detection algorithm based on improved SSD[J]. Infrared and Laser Engineering, 2018, 47(7): 0703005.

本文已被 7 篇论文引用
被引统计数据来源于中国光学期刊网
引用该论文: TXT   |   EndNote

相关论文

加载中...

关于本站 Cookie 的使用提示

中国光学期刊网使用基于 cookie 的技术来更好地为您提供各项服务,点击此处了解我们的隐私策略。 如您需继续使用本网站,请您授权我们使用本地 cookie 来保存部分信息。
全站搜索
您最值得信赖的光电行业旗舰网络服务平台!