吉林大学学报(地球科学版) ›› 2023, Vol. 53 ›› Issue (4): 1313-1322.doi: 10.13278/j.cnki.jjuese.20220197

• 地球探测与信息技术 • 上一篇    

基于改进YOLOX的遥感影像目标检测算法

李美霖1,2,芮杰1,金飞1,刘智1,林雨准1   

  1. 1.中国人民解放军战略支援部队信息工程大学地理空间信息学院,郑州 450001
    2.61206部队,北京 100042
  • 收稿日期:2022-07-07 出版日期:2023-07-26 发布日期:2023-08-11
  • 作者简介:李美霖(1995—),女,硕士,主要从事遥感影像智能解译方面的研究,E-mail: 643365777@qq.com
  • 基金资助:
    国家自然科学基金项目(41601507)

Remote Sensing Image Target Detection Algorithm Based on Improved YOLOX

Li Meilin1, 2, Rui Jie1, Jin Fei1, Liu Zhi1, Lin Yuzhun1   

  1. 1. Institute of Geospatial Information, PLA Strategic Support Force Information Engineering University, Zhengzhou 450001, China
    2. 61206 Troops, Beijing 100042, China
  • Received:2022-07-07 Online:2023-07-26 Published:2023-08-11
  • Supported by:
    the National Natural Science Foundation of China (41601507)

摘要: 目标检测是遥感影像处理中一项基础性和常规性的工作。本文基于YOLOX(you only look once X)网络进行改进,设计了一种针对遥感影像目标的检测算法。首先在特征提取模块PANet(path aggregation network)中加入自适应空间特征融合(adaptively spatial feature fusion, ASFF)网络,针对目标检测中尺度不一致的细部特征进行深入挖掘。其次,设计了基于ECA(efficient channel attention)的特征提取模块,高效通道交互在更加关注特征图中正样本特征信息的同时降低了模型的复杂性。再次,为了避免过拟合造成梯度消失、激活效果弱的问题,提出使用swish激活函数。最后,在DOTA(dataset for object detection in aerial images)上进行实验,通过消融实验定性分析、通过对比实验定量验证了本文算法的最佳机制和有效性。结果表明:在添加ASFF和ECA机制并且优化swish激活函数的前提下,改进网络模型的全类平均正确率(mean average precision, mAP)达74.42%,较原始网络提升了12.75%;与当前应用广泛的目标检测算法Mobilenet-YOLOv4、YOLOv4、YOLOv5、YOLOX相比,本文算法实现了mAP 11.42%~17.84%的提升。

关键词: 摄影测量, 遥感, 目标检测, DOTA, ECA, ASFF

Abstract: Target detection is a fundamental and routine task in remote sensing image processing. In this paper, we design a target detection algorithm for remote sensing images based on the YOLOX network. Firstly, ASFF is added to the feature extraction module PANet to deeply mine the fine features with inconsistent scale in target detection. Secondly, an ECA-based feature extraction module is designed with efficient channel interaction to reduce the complexity of the model while paying more attention to the positive sample feature information in the feature map. Then, to avoid the problem of gradient disappearance and weak activation effect caused by overfitting, the use of swish activation function is proposed. Finally, experiments are conducted on DOTA to verify the best mechanism and effectiveness of the improved method through qualitative analyses of the ablation experiments and quantitative comparison experiments. With the addition of ASFF and ECA mechanisms and the optimization of the swish activation function, the improved network model achieves an mAP of 74.42%, an improvement of 12.75% over the original network. Compared with the current widely used target detection algorithms Mobilenet-YOLOv4, YOLOv4, YOLOv5 and YOLOX, the proposed  algorithm achieves an improvement of 11.42%-17.84% in mAP accuracy.

Key words: photogrammetry, remote sensing, target detection, DOTA, ECA, ASFF

中图分类号: 

  • P237
[1] 赵忠海, 乔锴, 孙景贵, 陈俊, 崔晓梦, 梁杉杉, Manirambona Alain Jospin. 黑龙江建边农场地区遥感蚀变异常提取方法[J]. 吉林大学学报(地球科学版), 2023, 53(4): 1275-1275.
[2] 闫佰忠, 李瑶, 秦光雄, 于开宁, 吴云霞, 王亚楠. 基于遥感技术的贵德盆地多元信息干热岩靶区预测[J]. 吉林大学学报(地球科学版), 2023, 53(4): 1288-1300.
[3] 芦彦霖, 张森, 汲姣. 低空无人机倾斜摄影测量测图精度实证[J]. 吉林大学学报(地球科学版), 2021, 51(6): 1921-1931.
[4] 戴均豪, 薛林福, 李忠潭, 桑学佳, 马建雄. 无人机遥感技术在甘肃北山地区地质填图中的应用[J]. 吉林大学学报(地球科学版), 2021, 51(6): 1908-1920.
[5] 王博帅, 蒲东川, 李婷婷, 牛雪峰. 基于多源遥感影像的长春市城市建成区提取[J]. 吉林大学学报(地球科学版), 2021, 51(4): 1284-1294.
[6] 王明常, 朱春宇, 陈学业, 王凤艳, 李婷婷, 张海明, 韩有文. 基于FPN Res-Unet的高分辨率遥感影像建筑物变化检测[J]. 吉林大学学报(地球科学版), 2021, 51(1): 296-306.
[7] 孙立影, 杨晨, 赵海士, 常志勇. 基于极限学习机的遥感地球化学反演模型[J]. 吉林大学学报(地球科学版), 2020, 50(6): 1929-1938.
[8] 吴燕清, 王世成, 丁园, 王文正, 余弘龙, 王青, 李杨. 内蒙古新城子盆地铀及多金属矿产勘查遥感应用[J]. 吉林大学学报(地球科学版), 2020, 50(6): 1917-1928.
[9] 王凤艳, 赵明宇, 王明常, 张旭晴, 周凯. 无人机摄影测量在矿山地质环境调查中的应用[J]. 吉林大学学报(地球科学版), 2020, 50(3): 866-874.
[10] 柳长源, 刘鹏, 毕晓君. 基于自适应可变滤镜的地类变化预测模型[J]. 吉林大学学报(地球科学版), 2019, 49(5): 1477-1485.
[11] 刘子侠, 陈剑平, 王凤艳, 韩东亮. 基于活动控制的岩体结构面几何信息快速获取[J]. 吉林大学学报(地球科学版), 2019, 49(4): 1192-1199.
[12] 贺金鑫, 姜天, 董永胜, 韩凯旭, 马宁, 熊玥. 基于Landsat 8的辽宁弓长岭区遥感蚀变信息提取[J]. 吉林大学学报(地球科学版), 2019, 49(3): 893-901.
[13] 吴云霞, 吕凤军, 邢立新, 刘新星. 独山城地区多元信息干热岩预测模型[J]. 吉林大学学报(地球科学版), 2019, 49(3): 880-892.
[14] 樊瑞雪, 邢立新, 潘军, 单玄龙, 仲伟敬. 油砂的光谱反射特征及其遥感应用[J]. 吉林大学学报(地球科学版), 2019, 49(2): 603-610.
[15] 李刚, 孙桂华, 姚永坚, 朱博勤, 张耀明. 三沙湾海岸线时空演变[J]. 吉林大学学报(地球科学版), 2019, 49(1): 196-205.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
[1] 程立人,张予杰,张以春. 西藏申扎地区奥陶纪鹦鹉螺化石[J]. J4, 2005, 35(03): 273 -0282 .
[2] 李 秉 成. 陕西富平全新世古气候的初步研究[J]. J4, 2005, 35(03): 291 -0295 .
[3] 和钟铧,杨德明,王天武,郑常青. 冈底斯带巴嘎区二云母花岗岩SHRIMP锆石U-Pb定年[J]. J4, 2005, 35(03): 302 -0307 .
[4] 陈 力,佴 磊,王秀范,李 金. 绥中某电力设备站场区地震危险性分析[J]. J4, 2005, 35(05): 641 -645 .
[5] 纪宏金,孙丰月,陈满,胡大千,时艳香,潘向清. 胶东地区裸露含金构造的地球化学评价[J]. J4, 2005, 35(03): 308 -0312 .
[6] 初凤友,孙国胜,李晓敏,马维林,赵宏樵. 中太平洋海山富钴结壳生长习性及控制因素[J]. J4, 2005, 35(03): 320 -0325 .
[7] 李斌,孟自芳,李相博,卢红选,郑民. 泌阳凹陷下第三系构造特征与沉积体系[J]. J4, 2005, 35(03): 332 -0339 .
[8] 李涛, 吴胜军,蔡述明,薛怀平,YASUNORI Nakayama. 涨渡湖通江前后调蓄能力模拟分析[J]. J4, 2005, 35(03): 351 -0355 .
[9] 旷理雄,郭建华,梅廉夫,童小兰,杨丽. 从油气勘探的角度论博格达山的隆升[J]. J4, 2005, 35(03): 346 -0350 .
[10] 章光新,邓伟,何岩,RAMSIS Salama. 水文响应单元法在盐渍化风险评价中的应用[J]. J4, 2005, 35(03): 356 -0360 .