计算机科学
分割
人工智能
变压器
模式识别(心理学)
特征提取
特征学习
像素
工程类
电压
电气工程
作者
Shunli Wang,Qingwu Hu,Shaohua Wang,Qingwu Hu,Jiayuan Li,Mingyao Ai
出处
期刊:International journal of applied earth observation and geoinformation
日期:2024-03-01
卷期号:127: 103661-103661
标识
DOI:10.1016/j.jag.2024.103661
摘要
The semantic segmentation task is an essential issue in various fields, including land cover classification and cultural heritage investigation. The CNN and Transformer have been widely utilized in semantic segmentation tasks due to notable advancements in deep learning technologies. However, these methodologies may not fully account for remote sensing images' distinctive attributes, including the large intra-class variation and the small inter-class variation. Driven by it, we propose a category attention guided network (CAGNet). Initially, a local feature extraction module is devised to cater to striped objects and features at different scales. Then, we propose a novel concept of category attention for remote sensing images as a feature representation of category differences between pixels. Meanwhile, we designed the Transformer-based and CNN-based category attention guided modules to integrate the proposed category attention into the global scoring functions and local category feature weights, respectively. The network is designed to give more attention to the category features by updating these weights during the training process. Finally, a feature fusion module is developed to integrate global, local, and category multi-scale features and contextual information. A series of extensive experiments along with ablation studies on the UAVid, Vaihingen, and Potsdam datasets indicate that our network outperforms existing methods, including those based on CNN and Transformer.
科研通智能强力驱动
Strongly Powered by AbleSci AI