Research on Improved U-net Based Remote Sensing Image Segmentation Algorithm

Qiming Yang,Zixin Wang,Shinan Liu,Zizheng Li
2024-08-23
Abstract:In recent years, although U-Net network has made significant progress in the field of image segmentation, it still faces performance bottlenecks in remote sensing image segmentation. In this paper, we innovatively propose to introduce SimAM and CBAM attention mechanism in U-Net, and the experimental results show that after adding SimAM and CBAM modules alone, the model improves 17.41% and 12.23% in MIoU, and the Mpa and Accuracy are also significantly improved. And after fusing the two,the model performance jumps up to 19.11% in MIoU, and the Mpa and Accuracy are also improved by 16.38% and 14.8% respectively, showing excellent segmentation accuracy and visual effect with strong generalization ability and robustness. This study opens up a new path for remote sensing image segmentation technology and has important reference value for algorithm selection and improvement.
Distributed, Parallel, and Cluster Computing,Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
This paper aims to address the performance bottleneck in remote sensing image segmentation technology. Although the U-Net network has made significant progress in the field of image segmentation, there are still some challenges in remote sensing image segmentation. The paper proposes an innovative approach by introducing SimAM and CBAM attention mechanisms into the U-Net network to enhance the network's ability to capture and utilize key features of the image, thereby improving the performance of remote sensing image segmentation. Experimental results show that after adding the SimAM module alone, the model's mean Intersection over Union (MIoU) increased by 17.41%, while adding the CBAM module alone increased it by 12.23%. Further integrating both modules resulted in an MIoU improvement to 19.11%, and other metrics such as mean pixel accuracy (MPA) and overall accuracy also showed significant improvement. This indicates that the proposed method not only improves segmentation accuracy but also enhances the model's generalization ability and robustness. This research opens new avenues for remote sensing image segmentation technology and provides important reference value for algorithm selection and improvement.