Real-time road scene segmentation based on knowledge distillation Real-time road semantic segmentation

被引:0
作者
Li, Wenting [1 ]
Yang, Huicheng [1 ]
Hu, Yaocong [1 ]
Lin, Yuanyuan [1 ]
Shuai, Zhen [1 ]
机构
[1] Anhui Polytech Univ, Coll Elect Engn, Wuhu, Anhui, Peoples R China
来源
PROCEEDINGS OF 2023 7TH INTERNATIONAL CONFERENCE ON ELECTRONIC INFORMATION TECHNOLOGY AND COMPUTER ENGINEERING, EITCE 2023 | 2023年
关键词
Knowledge distillation; Attention mechanism; Asymmetric convolution; Real-time semantic segmentation;
D O I
10.1145/3650400.3650470
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Aiming at the problem that it is difficult to balance the accuracy and computational efficiency of the road scene semantic segmentation algorithm, a real-time road scene segmentation algorithm based on knowledge distillation is proposed. The algorithm introduces three key functional modules; (1) using an asymmetric convolutional pyramid module on the top of the encoder; (2) incorporating a coordinate attention module in the network; (3) Using high accuracy non-real time models to distill knowledge on lightweight model. The method in this paper achieves a balance of accuracy and realtime performance on the CamVid dataset. the inference speed is 63.7 FPS and the GFLOPs are 109.2 while the mIoU is 79.3% when computed on an NVIDIA GTX 1080Ti.
引用
收藏
页码:429 / 433
页数:5
相关论文
共 13 条
[1]  
Chen LC, 2017, Arxiv, DOI arXiv:1706.05587
[2]   DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFs [J].
Chen, Liang-Chieh ;
Papandreou, George ;
Kokkinos, Iasonas ;
Murphy, Kevin ;
Yuille, Alan L. .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2018, 40 (04) :834-848
[3]  
Liang-Chieh C, 2015, INT C LEARN REPR ICL
[4]   Real-Time Semantic Image Segmentation with Deep Learning for Autonomous Driving: A Survey [J].
Papadeas, Ilias ;
Tsochatzidis, Lazaros ;
Amanatiadis, Angelos ;
Pratikakis, Ioannis .
APPLIED SCIENCES-BASEL, 2021, 11 (19)
[5]  
Paszke A, 2016, Arxiv, DOI arXiv:1606.02147
[6]   FASSD-Net: Fast and Accurate Real-Time Semantic Segmentation for Embedded Systems [J].
Rosas-Arias, Leonel ;
Benitez-Garcia, Gibran ;
Portillo-Portillo, Jose ;
Olivares-Mercado, Jesus ;
Sanchez-Perez, Gabriel ;
Yanai, Keiji .
IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2022, 23 (09) :14349-14360
[7]   Fully Convolutional Networks for Semantic Segmentation [J].
Shelhamer, Evan ;
Long, Jonathan ;
Darrell, Trevor .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2017, 39 (04) :640-651
[8]  
Simonyan K, 2015, Arxiv, DOI arXiv:1409.1556
[9]  
Takos G., 2020, arXiv
[10]  
Wang HC, 2021, Arxiv, DOI arXiv:2102.04604