TY  - EJOU
AU  - Sun, Chuanlong 
AU  - Zhao, Hong 
AU  - Mu, Liang 
AU  - Xu, Fuliang 
AU  - Lu, Laiwei 

TI  - Image Semantic Segmentation for Autonomous Driving Based on Improved U-Net
T2  - Computer Modeling in Engineering \& Sciences

PY  - 2023
VL  - 136
IS  - 1
SN  - 1526-1506

AB  - Image semantic segmentation has become an essential part of autonomous driving. To further improve the generalization ability and the robustness of semantic segmentation algorithms, a lightweight algorithm network based on Squeeze-and-Excitation Attention Mechanism (SE) and Depthwise Separable Convolution (DSC) is designed. Meanwhile, Adam-GC, an Adam optimization algorithm based on Gradient Compression (GC), is proposed to improve the training speed, segmentation accuracy, generalization ability and stability of the algorithm network. To verify and compare the effectiveness of the algorithm network proposed in this paper, the trained network model is used for experimental verification and comparative test on the Cityscapes semantic segmentation dataset. The validation and comparison results show that the overall segmentation results of the algorithm network can achieve 78.02% MIoU on Cityscapes validation set, which is better than the basic algorithm network and the other latest semantic segmentation algorithms network. Besides meeting the stability and accuracy requirements, it has a particular significance for the development of image semantic segmentation.
KW  - Deep learning; semantic segmentation; attention mechanism; depthwise separable convolution; gradient compression

DO  - 10.32604/cmes.2023.025119