Ground-Based Remote Sensing Cloud Image Segmentation Using Convolution-MLP Network

Recently, MLPs-based methods in computer vision have attracted much attention due to the ability of learning long-range dependencies. However, MLPs-based methods usually treat all the tokens equally, which is difficult to segment challenging cloud regions. In this paper, we propose a novel network named Convolution-MLP Network (Con-MLPNet) for ground-based remote sensing cloud image segmentation, which could effectively learn long-range dependencies via the combination of MLPs and the attention mechanism. To this end, we propose the Attention Guided MLPs (AGMs) module to highlight salient features and suppress irrelevant features from the spatial and channel aspects. Meanwhile, different from existing MLPs methods where the long-range dependencies are learned from one single scale, we propose the Dilated MLPs (DMLPs) to learn long-range dependencies at different scales by sampling different channels of tokens. Furthermore, we design the Parallel Dilated MLPs (PDMs) module to integrate multiple DMLPs with different parameters in order to extract multi-scale information. We conduct a series of experiments on three public ground-based cloud image segmentation datasets, i.e., TLCDD, SWIMSEG and TCDD, and the results demonstrate that the proposed Con-MLPNet achieves state-of-the-art performance. Specifically, on the TLCDD dataset, our method surpasses the competing method across all five evaluation metrics, with the improvements of 3.3% in Precision, 2.48% in Recall, 3.74% in F-score, 1.76% in Accuracy, and 4.0% in IoU over the second-best results.