MSA-YOLO: A Remote Sensing Object Detection Model Based on Multi-Scale Strip Attention.

Zihang Su, Jiong Yu, Haotian Tan, Xueqiang Wan, Kaiyang Qi
Author Information
  1. Zihang Su: School of Software, Xinjiang University, Urumqi 830091, China.
  2. Jiong Yu: School of Software, Xinjiang University, Urumqi 830091, China.
  3. Haotian Tan: College of Information Science and Engineering, Xinjiang University, Urumqi 830046, China.
  4. Xueqiang Wan: School of Software, Xinjiang University, Urumqi 830091, China.
  5. Kaiyang Qi: School of Software, Xinjiang University, Urumqi 830091, China.

Abstract

Remote sensing image object detection holds significant research value in resources and the environment. Nevertheless, complex background information and considerable size differences between objects in remote sensing images make it challenging. This paper proposes an efficient remote sensing image object detection model (MSA-YOLO) to improve detection performance. First, we propose a Multi-Scale Strip Convolution Attention Mechanism (MSCAM), which can reduce the introduction of background noise and fuse multi-scale features to enhance the focus of the model on foreground objects of various sizes. Second, we introduce the lightweight convolution module GSConv and propose an improved feature fusion layer, which makes the model more lightweight while improving detection accuracy. Finally, we propose the Wise-Focal CIoU loss function, which can reweight different samples to balance the contribution of different samples to the loss function, thereby improving the regression effect. Experimental results show that on the remote sensing image public datasets DIOR and HRRSD, the performance of our proposed MSA-YOLO model is significantly better than other existing methods.

Keywords

References

  1. IEEE Trans Pattern Anal Mach Intell. 2017 Jun;39(6):1137-1149 [PMID: 27295650]
  2. IEEE Trans Pattern Anal Mach Intell. 2020 Feb;42(2):386-397 [PMID: 29994331]
  3. Sensors (Basel). 2022 May 20;22(10): [PMID: 35632300]
  4. IEEE Trans Pattern Anal Mach Intell. 2020 Feb;42(2):318-327 [PMID: 30040631]
  5. IEEE Trans Pattern Anal Mach Intell. 2020 Aug;42(8):2011-2023 [PMID: 31034408]

Grants

  1. 62262064,62266043,61966035/National Natural Science Foundation of China Project under Grant
  2. XJEDU2016S106/Key R&D projects in Xinjiang Uygur Autonomous Region
  3. 2022D01C56/Natural Science Foundation of Xinjiang Uygur Autonomous Region of China

Word Cloud

Created with Highcharts 10.0.0sensingdetectionremotemodelimageobjectproposeRemotebackgroundobjectsimagesMSA-YOLOperformanceMulti-ScaleStripAttentioncanlightweightimprovinglossfunctiondifferentsamplesholdssignificantresearchvalueresourcesenvironmentNeverthelesscomplexinformationconsiderablesizedifferencesmakechallengingpaperproposesefficientimproveFirstConvolutionMechanismMSCAMreduceintroductionnoisefusemulti-scalefeaturesenhancefocusforegroundvarioussizesSecondintroduceconvolutionmoduleGSConvimprovedfeaturefusionlayermakesaccuracyFinallyWise-FocalCIoUreweightbalancecontributiontherebyregressioneffectExperimentalresultsshowpublicdatasetsDIORHRRSDproposedsignificantlybetterexistingmethodsMSA-YOLO:SensingObjectDetectionModelBasedYOLOnetworkattentionmechanism

Similar Articles

Cited By