Satellite Image for Cloud and Snow Recognition Based on Lightweight Feature Map Attention Network

被引:2
作者
Yang, Chaoyun [1 ]
Zhang, Yonghong [1 ,2 ]
Xia, Min [1 ,2 ]
Lin, Haifeng [3 ]
Liu, Jia [1 ]
Li, Yang [1 ]
机构
[1] Nanjing Univ Informat Sci & Technol, Collaborat Innovat Ctr Atmospher Environm & Equip, Nanjing 210044, Peoples R China
[2] Nanjing Univ Informat Sci & Technol, Jiangsu Key Lab Big Data Anal Technol, Nanjing 210044, Peoples R China
[3] Nanjing Forestry Univ, Coll Informat Sci & Technol, Nanjing 210037, Peoples R China
基金
中国国家自然科学基金;
关键词
cloud and snow recognition; convolutional neural network; lightweight feature map; attention network; REMOTE-SENSING IMAGES;
D O I
10.3390/ijgi11070390
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Cloud and snow recognition technology is of great significance in the field of meteorology, and is also widely used in remote sensing mapping, aerospace, and other fields. Based on the traditional method of manually labeling cloud-snow areas, a method of labeling cloud and snow areas using deep learning technology has been gradually developed to improve the accuracy and efficiency of recognition. In this paper, from the perspective of designing an efficient and lightweight network model, a cloud snow recognition model based on a lightweight feature map attention network (Lw-fmaNet) is proposed to ensure the performance and accuracy of the cloud snow recognition model. The model is improved based on the ResNet18 network with the premise of reducing the network parameters and improving the training efficiency. The main structure of the model includes a shallow feature extraction module, an intrinsic feature mapping module, and a lightweight adaptive attention mechanism. Overall, in the experiments conducted in this paper, the accuracy of the proposed cloud and snow recognition model reaches 95.02%, with a Kappa index of 93.34%. The proposed method achieves an average precision rate of 94.87%, an average recall rate of 94.79%, and an average F1-Score of 94.82% for four sample recognition classification tasks: no snow and no clouds, thin cloud, thick cloud, and snow cover. Meanwhile, our proposed network has only 5.617M parameters and takes only 2.276 s. Compared with multiple convolutional neural networks and lightweight networks commonly used for cloud and snow recognition, our proposed lightweight feature map attention network has a better performance when it comes to performing cloud and snow recognition tasks.
引用
收藏
页数:23
相关论文
共 39 条
[1]  
[Anonymous], 2017, P IEEE C COMP VIS PA
[2]  
Ashraf K., 2016, SQUEEZENET ALEXNET L
[3]   Cloud Detection for High-Resolution Satellite Imagery Using Machine Learning and Multi-Feature Fusion [J].
Bai, Ting ;
Li, Deren ;
Sun, Kaimin ;
Chen, Yepei ;
Li, Wenzhuo .
REMOTE SENSING, 2016, 8 (09)
[4]   MANet: a multi-level aggregation network for semantic segmentation of high-resolution remote sensing images [J].
Chen, Bingyu ;
Xia, Min ;
Qian, Ming ;
Huang, Junqing .
INTERNATIONAL JOURNAL OF REMOTE SENSING, 2022, 43 (15-16) :5874-5894
[5]   Multilevel Cloud Detection for High-Resolution Remote Sensing Imagery Using Multiple Convolutional Neural Networks [J].
Chen, Yang ;
Fan, Rongshuang ;
Bilal, Muhammad ;
Yang, Xiucheng ;
Wang, Jingxue ;
Li, Wei .
ISPRS INTERNATIONAL JOURNAL OF GEO-INFORMATION, 2018, 7 (05)
[6]   MLNet: multichannel feature fusion lozenge network for land segmentation [J].
Gao, Jiahong ;
Weng, Liguo ;
Xia, Min ;
Lin, Haifeng .
JOURNAL OF APPLIED REMOTE SENSING, 2022, 16 (01)
[7]   Res2Net: A New Multi-Scale Backbone Architecture [J].
Gao, Shang-Hua ;
Cheng, Ming-Ming ;
Zhao, Kai ;
Zhang, Xin-Yu ;
Yang, Ming-Hsuan ;
Torr, Philip .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2021, 43 (02) :652-662
[8]   GhostNet: More Features from Cheap Operations [J].
Han, Kai ;
Wang, Yunhe ;
Tian, Qi ;
Guo, Jianyuan ;
Xu, Chunjing ;
Xu, Chang .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2020, :1577-1586
[9]   Channel Pruning for Accelerating Very Deep Neural Networks [J].
He, Yihui ;
Zhang, Xiangyu ;
Sun, Jian .
2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2017, :1398-1406
[10]  
Howard A. G., 2017, arXiv