Improved Sea Ice Image Segmentation using U2-Net and Data Set Augmentation
Preprint
- 17 July 2023
- preprint
- Published by MDPI AG in Preprints.org
Abstract
Sea ice extraction and segmentation of remote sensing images is the basis for sea ice monitoring. Machine learning-based image segmentation methods rely on manual sampling and require complex feature extraction. Deep-learning semantic segmentation methods have the advantages of high efficiency, intelligence, and automation. Sea ice segmentation using deep learning methods faces the following problems: in terms of datasets, the high cost of sea ice image label production leads to fewer datasets for sea ice segmentation; in terms of image quality, remote sensing image noise and Severe weather conditions affects image quality, which affects the ac-curacy of sea ice extraction. To address the quantity and quality of the dataset, this study used multiple data augmentation methods for data expansion. To improve the semantic segmentation accuracy, the SC-U2-Net network was constructed using multi-scale inflation convolution and a multi-layer Convolutional Block Attention Module (CBAM) attention mechanism for the U2-Net network. The experiments showed that (1) data augmentation solved the problem of an insuffi-cient number of training samples to a certain extent and improved the accuracy of image seg-mentation. (2) This study designed a multilevel Gaussian noise data augmentation scheme to improve the network's ability to resist noise interference and achieve a more accurate segmenta-tion of images with different degrees of noise pollution. (3) The inclusion of a multi-scale inflation perceptron and multi-layer CBAM attention mechanism improved the ability of U2-Net network feature extraction and enhanced the model accuracy and generalization ability.Keywords
All Related Versions
- Version 1, 2023-07-17, Preprints
- Published version: Applied Sciences, 13 (16), 9402.
This publication has 13 references indexed in Scilit:
- Performance Analysis of the YOLOv4 Algorithm for Pavement Damage Image Detection with Different Embedding Positions of CBAM ModulesApplied Sciences, 2022
- Cross-scenario transfer diagnosis of reciprocating compressor based on CBAM and ResNetJournal of Intelligent & Fuzzy Systems: Applications in Engineering and Technology, 2022
- Remote Sensing Sea Ice Image Classification Based on Multilevel Feature Fusion and Residual NetworkMathematical Problems in Engineering, 2021
- Sea Ice Image Classification Based on Heterogeneous Data Fusion and Deep LearningRemote Sensing, 2021
- An Efficient Module for Instance Segmentation Based on Multi-Level Features and Attention MechanismsApplied Sciences, 2021
- Sea Ice Classification via Deep Neural Network Semantic SegmentationIEEE Sensors Journal, 2020
- U2-Net: Going deeper with nested U-structure for salient object detectionPattern Recognition, 2020
- Hyperspectral Sea Ice Image Classification Based on the Spectral-Spatial-Joint Feature with Deep LearningRemote Sensing, 2019
- CBAM: Convolutional Block Attention Module2018
- A hardware Gaussian noise generator using the Box-Muller method and its error analysisIEEE Transactions on Computers, 2006