基于改进DeepLab V3+网络的无人机影像语义分割方法OA
Segmentation Method for UAV Imagery Based on Improved DeepLab V3+
针对无人机影像目标尺度差异大、样本分布不均等特点带来的语义分割效果差且分割边界模糊的问题,文章提出了一种改进的DeepLab V3+语义分割方法.所提方法的改进之处包括:1)设计了一种融合通道注意力(Channel Attention,CA)与位置注意力(Position Attention,PA)的CA-PA注意力机制,增强了模型对关键信息和多尺度目标的特征提取能力;2)引入FFM(Feature Fusion Module)特征融合模块,实现了低级与高级特征的高效融合,改善了边界模糊和小目标分割问题;3)采用Dice Loss与Focal Loss的加权混合损失函数,减小了类别不平衡和分类困难样本的影响.基于Aeroscapes和UAVid数据集的测试结果表明,相较于现有语义分割网络,所提方法提升了多尺度目标以及分布不均样本的处理能力,能够对边界进行精确分割,更适用于无人机影像的语义分割任务.
To address the challenges of poor semantic segmentation performance and blurred boundaries caused by sig-nificant differences in target scales and uneven sample distribution in UAV imagery,an improved DeepLab V3+seman-tic segmentation network is proposed.The proposed method includes the following enhancements:1)a CA-PA attention mechanism is designed to improve the model's ability to extract critical information and features from multi-scale tar-gets;2)an FFM feature fusion module is introduced to effectively integrate low-level and high-level features,thereby al-leviating boundary blurriness and enhancing the segmentation of small targets;and 3)a weighted hybrid loss function combining Dice Loss and Focal Loss is adopted to address class imbalance and improve the learning of challenging samples.Experimental results based on the Aeroscapes and UAVid datasets show that compared to existing semantic segmentation networks,the proposed method improves the processing ability of multi-scale targets and imbalanced samples,and achieves more accurate boundary segmentation,making it highly suitable for UAV imagery semantic seg-mentation tasks.
苏楠;赵振华
南京航空航天大学自动化学院,江苏 南京 211106南京航空航天大学自动化学院,江苏 南京 211106
信息技术与安全科学
语义分割DeepLab V3+注意力机制特征融合混合损失函数
semantic segmentationDeepLab V3+attention mechanismfeature fusionhybrid loss function
《海军航空大学学报》 2026 (3)
547-556,10
江苏省自然科学基金优秀青年基金(BK20230091)国家自然科学基金(62573230)
评论