改进PP-LiteSeg的轻量级无人机影像语义分割算法OA
An Algorithm of Segmenting Lightweight Drone Image Semanteme Based on Improved PP-LiteSeg
针对现有语义分割算法在无人机航拍影像的重点区域检测中存在的分割精度低、检测速度慢的问题,提出了一种改进PP-LiteSeg的轻量级无人机影像语义分割算法.该算法首先设计了一种复合注意力融合模块,在统一注意力融合模块中引入无参数注意力机制SimAM,增强全局上下文信息,提升输出特征的信息丰富度;之后,通过将主干网络中卷积的计算方式由普通卷积替换为部分卷积与小尺度卷积核结合的方式,减少模型参数;同时,设计了新的主干网络SDTCM_PNet,通过更改主干网络中短期密集连接模块在多层感受野下的特征拼接方式,进一步提升模型的轻量化程度.在自采的无人机航拍影像数据集上进行的实验结果证明了本文算法的有效性.同时将算法在嵌入式设备上进行了部署试验,结果验证了本文算法满足实时性要求.
In response to the problems that segmentation is low in accuracy and detection is slow at speed in detecting drone aerial images in key areas by using existing semantic segmentation algorithm,an im-proved PP-LiteSeg lightweight drone image semantic segmentation algorithm is proposed.The algorithm,first,is to design a composite attention fusion module in which the parameter free attention mechanism Si-mAM is introduced into the unified attention fusion module to enhance global contextual information and improve the information richness of output features.Afterwards,the model parameters are reduced through replacing the calculation method of convolution in the backbone network from ordinary convolu-tion to a combination of partial convolution and small-scale convolution kernels.At the same time,a new backbone network SDTCM_PNet is designed to further enhance the lightweighting of the model by chan-ging the feature concatenation method of short-term dense connection modules in the multi-layer receptive field of the backbone network.The experimental results conducted on the self-collected drone aerial image dataset show that the algorithm proposed in this paper is valid.Simultaneously,the algorithm is to be de-ployed and tested on embedded devices,and the algorithm also meets the needs of real-time.
李浩;贺云涛;李子豪
北京理工大学空天科学与技术学院,北京,100081北京理工大学空天科学与技术学院,北京,100081中国航天科技集团第十一研究院,北京,100074
航空航天
语义分割轻量化无人机影像无参数注意力重点区域检测
semantic segmentationlightweightdrone imageparameter free attentionkey area detection
《空军工程大学学报》 2026 (1)
21-31,11
航空科学基金(2020Z005072001)
评论