A serial semantic segmentation model based on encoder-decoder architecture

被引:2
作者
Zhou, Yan [1 ,2 ]
机构
[1] Zhejiang Univ, Ocean Coll, Zhoushan 316021, Peoples R China
[2] Minist Nat Resources, Inst Oceanog 2, State Key Lab Satellite Ocean Environm Dynam, Hangzhou 310012, Peoples R China
关键词
Semantic segmentation; Lawin transformer; CNN; Encoder-decoder; Attention mechanism; DEEP CONVOLUTIONAL NETWORKS; FUSION NETWORK;
D O I
10.1016/j.knosys.2024.111819
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The thriving progress of Convolutional Neural Networks (CNNs) and the outstanding efficacy of Visual Transformers (ViTs) have delivered impressive outcomes in the domain of semantic segmentation. However, each model in isolation entails a trade-off between high computational complexity and compromised computational efficiency. To address this challenge, we effectively combine the CNN and encoder-decoder structures in a Transformer-inspired fashion, presenting the Serial Semantic Segmentation Trans via CNN Former (SSS-Former) model. To augment the feature extraction capability, we utilize the meticulously crafted SSS-CSPNet, resulting in a well-designed architecture for the holistic model. We propose a novel SSS-PN attention network that enhances the spatial topological connections of features, leading to improved overall performance. Additionally, the integration of SASPP bridges the semantic gap between multi-scale features and enhances segmentation ability for overlapping objects. To fulfill the requirement of real-time segmentation, we leverage a novel restructuring technique to devise a more lightweight and faster ResSSS-Former model. Abundant experimental results demonstrate that both SSS-Former and ResSSS-Former outperform existing state-of-the-art methods in terms of computational efficiency, result precision, and speed. Remarkably, SSS-Former achieves a mIoU of 58.63 % at 89.1FPS on the ADE20K dataset. On the validation and testing datasets of CityScapes, it obtains mIoU scores of 85.1 % and 85.2 % respectively, with a speed of 94.1FPS. Our optimized ResSSS-Former achieves impressive realtime segmentation results, with an astonishing 100+FPS while maintaining high segmentation accuracy. The compelling results from the ISPRS datasets further validate the effectiveness of our proposed models in segmenting multi-scale and overlapping objects.
引用
收藏
页数:18
相关论文
共 80 条
[71]   ShuffleNet: An Extremely Efficient Convolutional Neural Network for Mobile Devices [J].
Zhang, Xiangyu ;
Zhou, Xinyu ;
Lin, Mengxiao ;
Sun, Ran .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :6848-6856
[72]   Pyramid Scene Parsing Network [J].
Zhao, Hengshuang ;
Shi, Jianping ;
Qi, Xiaojuan ;
Wang, Xiaogang ;
Jia, Jiaya .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :6230-6239
[73]  
Zhao X, 2023, Arxiv, DOI arXiv:2306.12156
[74]   Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformers [J].
Zheng, Sixiao ;
Lu, Jiachen ;
Zhao, Hengshuang ;
Zhu, Xiatian ;
Luo, Zekun ;
Wang, Yabiao ;
Fu, Yanwei ;
Feng, Jianfeng ;
Xiang, Tao ;
Torr, Philip H. S. ;
Zhang, Li .
2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, :6877-6886
[75]   Scene Parsing through ADE20K Dataset [J].
Zhou, Bolei ;
Zhao, Hang ;
Puig, Xavier ;
Fidler, Sanja ;
Barriuso, Adela ;
Torralba, Antonio .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :5122-5130
[76]   Joint 3D Instance Segmentation and Object Detection for Autonomous Driving [J].
Zhou, Dingfu ;
Fang, Jin ;
Song, Xibin ;
Liu, Liu ;
Yin, Junbo ;
Dai, Yuchao ;
Li, Hongdong ;
Yang, Ruigang .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2020, :1836-1846
[77]   A YOLO-NL object detector for real-time detection [J].
Zhou, Yan .
EXPERT SYSTEMS WITH APPLICATIONS, 2024, 238
[78]   IYOLO-NL: An improved you only look once and none left object detector for real-time face mask detection [J].
Zhou, Yan .
HELIYON, 2023, 9 (08)
[79]   Self-attention feature fusion network for semantic segmentation [J].
Zhou, Zhen ;
Zhou, Yan ;
Wang, Dongli ;
Mu, Jinzhen ;
Zhou, Haibin .
NEUROCOMPUTING, 2021, 453 :50-59
[80]   A Good Student is Cooperative and Reliable: CNN-Transformer Collaborative Learning for Semantic Segmentation [J].
Zhu, Jinjing ;
Luo, Yunhao ;
Zheng, Xu ;
Wang, Hao ;
Wang, Lin .
2023 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2023), 2023, :11686-11696