电光与控制, 2022, 29 (8): 45, 网络出版: 2022-08-14   

基于深度可分离卷积的实时遥感目标检测算法

A Real-Time Remote Sensing Target Detection Algorithm Based on Depth Separable Convolution
作者单位
上海电力大学电子与信息工程学院, 上海 201000
摘要
针对基于深度学习的遥感目标检测算法参数冗余、计算量大且实时检测性能较差的问题, 提出了一种基于深度可分离卷积的实时遥感目标检测算法。首先通过K-means++算法对数据集进行锚框(Anchor)聚类分析, 使锚框参数更加符合遥感检测场景。为了降低模型参数量、提升检测速度, 以轻量级网络MobileNetv3作为主干网络进行特征提取; 此外, 基于深度可分离卷积的PANet(Path Aggregation Network)结构的设计, 使网络参数量进一步降低。改进后模型参数量仅为原来的18.3%, 检测速度提升2.19倍, 在UCAS_AOD, RSOD, DIOR这3个遥感数据集上进行测试, 实验结果表明, 算法鲁棒性强, 能够在保证模型检测精度的同时有效提高检测实时性。
Abstract
Remote sensing target detection algorithms based on deep learning have redundant parameters, large computation amount and poor real-time detection performance.To solve the above problems, a real-time remote sensing target detection algorithm based on depth separable convolution is proposed.The datasets are analyzed by anchor box (Anchor) clustering via the K-means++ algorithm, making the anchor box parameters more compliant with the remote sensing detection scenario.In order to reduce the quantity of model parameters and improve the detection speed, feature extraction is performed with the lightweight network MobileNetv3 as the backbone network.In addition, the design of PANet (Path Aggregation Network) structure based on depth separable convolution makes the quantity of network parameters further reduced.The quantity of model parameters after improvement is only 18.3% of the original, and the detection speed is 2.19 times faster than the original.Tests are conducted on three remote sensing datasets, that is, UCAS_AOD, RSOD and DIOR, and the experimental results show that the algorithm is robust and can effectively improve the real-time detection performance while ensuring the model detection accuracy.

王成龙, 赵倩, 赵琰, 郭彤. 基于深度可分离卷积的实时遥感目标检测算法[J]. 电光与控制, 2022, 29(8): 45. WANG Chenglong, ZHAO Qian, ZHAO Yan, GUO Tong. A Real-Time Remote Sensing Target Detection Algorithm Based on Depth Separable Convolution[J]. Electronics Optics & Control, 2022, 29(8): 45.

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

相关论文

加载中...

关于本站 Cookie 的使用提示

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