SLWE-Net: An improved lightweight U-Net for Sargassum extraction from GOCI images

Mar Pollut Bull. 2023 Sep;194(Pt B):115349. doi: 10.1016/j.marpolbul.2023.115349. Epub 2023 Aug 7.

Abstract

The Sargassum bloom has severely impacted the ecological environment of the East China Sea and the Yellow Sea, causing significant economic losses. In recent years, deep learning has seen extensive development due to its outstanding feature extraction capabilities. However, the deep learning process typically involves a large number of parameters and computations. To address this issue, this paper proposes a lightweight deep learning network based on the U-Net framework, called SLWE-NET, which uses lightweight modules to replace the feature extraction modules in U-Net. In this experiment, SLWE-Net performed the best in both extraction accuracy and model lightweight. Compared to the formal U-Net, the number of parameters decreased by 65.83 %, the model size reduced from 94.97 MB to 32.51 MB, and the mIoU increased to 93.81 %. Therefore, the method proposed in this paper is beneficial for Sargassum extraction and provides a basis for operational monitoring.

Keywords: Deep learning; Geostationary ocean color imager (GOCI); Lightweight; Sargassum.

MeSH terms

  • China
  • Environment
  • Sargassum*