基于FNM-Net的轻量级遥感目标检测算法

打开文本图片集
Lightweight remote sensing object detection algorithm based on FNM-Net
WENBin²,ZHANG Jun1,2 ,WANGJunyin²,WANGZihao²,DINGYifu (1.HubeioalbodtfadropisUst; 2.SchoolofElectricaland NewEnergy,ChinaThreeGorgesUniversity,Yichang 443oO2,China; 3.NanchuanPowerSupply Branch of StateGridChongqing ElectricPowerCompany,Chongqing 4O4o,China)
Abstract:Inviewofthe current challnges inremote sensing object detection,such as low accuracy,slow processing speedsandlargequantityof modelparameters,anFNM-Net,alightweightremotesensingobject detectionnetwork basedonan improvedYOLOv7-tinyarchitecture,isproposed.Firstly,alightweightfeatureextractionnetworkFaster-Netisintroducedto substitutetheoriginalone,soastopreventthenetwork'sexcesiveoverlapoffeaturemaps.Secondlythefocalmodulation moduleisintroducedandthespatialinformationintegrationmodule (SIM)isproposed toconstructanewpathaggregationnetwork thatmitigatesissuesoftheinformationredundancyandtheoverlookof intra-layerfeaturesduringfeaturefusion.Andthen,the multi-fine-graineddetectionheadsaredesignedtoavoidthelargescalevariationsofremotesensingobjects.Finalyapruning method utilizinga layeradaptiveamplitudepruning (LAMP)scoreisemployed to eliminateconnections withminor weights, therebyreducingparameternumberandcomputationalburdenandincreasingthedetectionspeed.Thismethodisvalidatedbythe public RSOD dataset.The results show that the proposed method achieves a 51.2% reduction in parameter number,a 55.2% (20 decrease in FLOPs,a 6.5f/s increase in detection speed,and a 2.1% improvement in mAP (mean average precision) in comparisonwiththoseof thebaseline model.Additionaly,itsgeneralizabilityisconfirmedontheNWPU VHR-1Odataset.
Keywords:remote sensing object detection;FNM-Net; light weight; pruning; improvedYOLOv7-tiny; SIM
0 引言
遥感图像广泛应用于环境监测、矿产资源勘探、交通规划、灾害监测与救援、国防军事等多个领域。(剩余11290字)