Electronics Optics & Control, Volume. 30, Issue 1, 92(2023)
U-Net Semantic Segmentation with ECA Attention Mechanism
Many applications rely on the accuracy of data understanding, and semantic image segmentation effectively solves this problem, which provides the necessary context information for scene understanding based on pixel level.A U-Net network structure, ECAU-Net, is proposed based on ResNeXt50.Compared with the general convolution operation, ResNeXt50 has a stronger feature extraction ability.In fusion process, Efficient Channel Attention module(ECA-Net) is introduced to further enhance the ability of feature representation to discriminate scene segmentation.In addition, the introduction of dilated convolution in the overall network structure expands the receptive field of the image without changing the size of convolution kernel, thereby maximizing the performance of the network.The experimental results show that on the CamVid dataset, compared with U-Net, ECAU-Net has improved the Acc, Acc class, MIoU and FWIoU respectively by 2.1%, 8.6%, 8.2% and 3.2%.
Get Citation
Copy Citation Text
WANG Ruishen, SONG Gongfei, WANG Ming. U-Net Semantic Segmentation with ECA Attention Mechanism[J]. Electronics Optics & Control, 2023, 30(1): 92
Category:
Received: Oct. 21, 2021
Accepted: --
Published Online: Apr. 3, 2023
The Author Email: