Please wait a minute...
 
Remote Sensing for Natural Resources    2022, Vol. 34 Issue (2) : 88-96     DOI: 10.6046/zrzyyg.2021128
|
Cloud detection algorithm of remote sensing image based on DenseNet and attention mechanism
LIU Guangjin1,2(), WANG Guanghui1,2(), BI Weihua3, LIU Huijie2, YANG Huachao1
1. School of Environment and Spatial Informatics, China University of Mining and Technology, Xuzhou 221116, China
2. Land Satellite Remote Sensing Application Center, MNR, Beijing 100048, China
3. Wanbei Coal and Electricity Co.Ltd., Suzhou 234002, China
Download: PDF(6594 KB)   HTML
Export: BibTeX | EndNote | Reference Manager | ProCite | RefWorks    
Abstract  

The cloud detection of remote sensing images is the first step in the process of remote sensing image processing. To address the problem that the traditional cloud detection algorithm has a poor effect on the detection of small and thin clouds, this paper proposes a cloud detection method for densely connected network remote sensing images based on the attention mechanism. First, cloud vectors are manually checked from the images provided by the Land Satellite Remote Sensing Application Center of the Ministry of Natural Resources and cloud labels are made, and the images are preprocessed by sequential clipping, color jitter, rotation, etc. to enlarge the sample size. Then, the pre-processed remote sensing images and their labels are fed into a neural network with DenseNet as the encoder and decoder, and a cascaded atrous convolution module is added between the encoder and decoder to increase the receptive field, and a dual attention mechanism and a global context modeling module are added to suppress some irrelevant detailed information. Finally, the experimental results showed that the accuracy rate could reach 95% and the intersection over union could reach 91%, which are big improvements over the traditional cloud detection algorithm, and this method performs well in extracting small and thin clouds.

Keywords cloud detection      DenseNet      dual attention mechanism      global context modeling module      atrous convolution     
ZTFLH:  TP751.1  
Corresponding Authors: WANG Guanghui     E-mail: 1538868186@qq.com;wanggh@lasac.cn
Issue Date: 20 June 2022
Service
E-mail this article
E-mail Alert
RSS
Articles by authors
Guangjin LIU
Guanghui WANG
Weihua BI
Huijie LIU
Huachao YANG
Cite this article:   
Guangjin LIU,Guanghui WANG,Weihua BI, et al. Cloud detection algorithm of remote sensing image based on DenseNet and attention mechanism[J]. Remote Sensing for Natural Resources, 2022, 34(2): 88-96.
URL:  
https://www.gtzyyg.com/EN/10.6046/zrzyyg.2021128     OR     https://www.gtzyyg.com/EN/Y2022/V34/I2/88
Fig.1  ResNet-Block structure
Fig.2  Densely connected block
Fig.3  Transition block
Fig.4  Channel attention model
Fig.5  Position attention model
Fig.6  Non-local neural networks
Fig.7  Global context block
Fig.8  Dilated-convolution block
Fig.9  Attention DBlock densely connected networks
编号 原始影像 真值标签 影像类型
1 草地
2 荒地
3 水域
4 城镇
5 山地
Tab.1  Original image and its truth tag
编号 增强后影像 增强后影像的真值标签
1
2
3
编号 增强后影像 增强后影像的真值标签
4
5
Tab.2  Enhanced image and its truth tag
算法模型 Recall Precision IoU
Otsu阈值法 0.865 706 0.341 597 0.324 407
Otsu多阈值法 0.684 431 0.833 832 0.602 283
K-means聚类法 0.838 212 0.385 039 0.358 404
SegNet 0.908 152 0.936 953 0.855 852
Unet 0.945 295 0.954 285 0.904 343
D-LinkNet50 0.942 009 0.948 958 0.896 580
D-DenseNet 0.947 085 0.955 037 0.906 659
AD-DenseNet 0.948 931 0.957 636 0.910 701
Tab.3  Evaluation results of different algorithm models on training sets
算法模型 Recall Precision IoU
Otsu阈值法 0.859 295 0.389 704 0.366 328
Otsu多阈值法 0.702 932 0.868 563 0.635 349
K-means聚类法 0.841 581 0.423 713 0.392 414
SegNet 0.923 405 0.926 182 0.860 104
Unet 0.946 430 0.958 225 0.908 927
D-LinkNet50 0.948 262 0.949 005 0.902 285
D-DenseNet 0.951 485 0.958 911 0.904 211
AD-DenseNet 0.953 866 0.961 321 0.918 611
Tab.4  Evaluation results of different algorithm models on the validation sets
Fig.10  Change curve of AD-DenseNet algorithm loss and IoU with epoch
Fig.11  Change curve of AD-DenseNet algorithm precision and recall with epoch
Fig.12  Comparison of cloud detection results of different algorithms
[1] Zhang Y, Rossow W B, Lacis A, et al. Calculation of radiative fluxes from the surface to top of atmosphere based on ISCCP and other global data sets:Refinements of the radiative transfer model and the input data[J]. Journal of Geophysical Research:Atmospheres, 2004, 109(d19):105-115.
[2] Ju J C, Roy D P. The availability of cloud-free Landsat ETM+ data over the conterminous United States and globally[J]. Remote Sensing of Environment, 2007, 112(3):1196-1211.
doi: 10.1016/j.rse.2007.08.011 url: https://linkinghub.elsevier.com/retrieve/pii/S0034425707004002
[3] Xie F, Shi M Y, Shi Z, et al. Multilevel cloud detection in remote sensing images based on deep learning[J]. IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing, 2017, 10:3631-3640.
doi: 10.1109/JSTARS.2017.2686488 url: https://ieeexplore.ieee.org/document/7895175/
[4] Zhu Z, Woodcock C E. Object-based cloud and cloud shadow detection in Landsat imagery[J]. Remote Sensing of Environment, 2012, 118:83-94.
doi: 10.1016/j.rse.2011.10.028 url: https://linkinghub.elsevier.com/retrieve/pii/S0034425711003853
[5] Otsu N. A Threshold selection method from gray-level histograms[J]. IEEE Transactions on Systems Man and Cybernetics, 2007, 9(1):62-66.
doi: 10.1109/TSMC.1979.4310076 url: http://ieeexplore.ieee.org/document/4310076/
[6] 李玉功. K-means聚类算法的改进及其在彩色图像分割中的应用[D]. 合肥: 安徽大学, 2018.
[6] Li Y G. Improvement of K-means clustering algorithm and its application in color image segmentation[D]. Hefei: Anhui University, 2018.
[7] 康一飞, 潘励, 孙明伟, 等. 基于高斯混合模型法的国产高分辨率卫星影像云检测[J]. 武汉大学学报(信息科学版), 2017, 42(6):782-788.
[7] Kang Y F, Pan L, Sun M W, et al. Gaussian mixture model based cloud detection for Chinese high resolution satellite imagery[J]. Geomatics and Information Science of Wuhan University, 2017, 42(6):782-788.
[8] 仇一帆, 柴登峰. 无人工标注数据的Landsat影像云检测深度学习方法[J]. 国土资源遥感, 2021, 33(1):102-107.doi: 10.6046/gtzyyg.2020090.
doi: 10.6046/gtzyyg.2020090
[8] Qiu Y F, Chai D F. A deep learning method for Landsat image cloud detection without manually labeled data[J]. Remote Sensing for Land and Resources, 2021, 33(1):102-107.doi: 10.6046/gtzyyg.2020090.
doi: 10.6046/gtzyyg.2020090
[9] 栗旭升, 刘玉峰, 陈冬花, 等. 结合图像特征的支持向量机高分一号云检测[J]. 国土资源遥感, 2020, 32(3):55-62.doi: 10.6046/gtzyyg.2020.03.08.
doi: 10.6046/gtzyyg.2020.03.08
[9] Li X S, Liu Y F, Chen D H, et al. Cloud detection based on support vector machine with image features for GF-1 data[J]. Remote Sensing for Land and Resources, 2020, 32(3):55-62.doi: 10.6046/gtzyyg.2020.03.08.
doi: 10.6046/gtzyyg.2020.03.08
[10] 徐启恒, 黄滢冰, 陈洋. 结合超像素和卷积神经网络的国产高分辨率遥感影像云检测方法[J]. 测绘通报, 2019(1):50-55.
[10] Xu Q H, Huang Y B, Chen Y. Cloud detection for Chinese high resolution remote sensing imagery using combining superpixel with convolution neural network[J]. Bulletin of Surveying and Mapping, 2019(1):50-55.
[11] 刘云峰, 杨珍, 韩骁, 等. 国产高分辨率遥感卫星影像云检测方法分析[J]. 测绘通报, 2020(11):66-70.
[11] Liu Y F, Yang Z, Han X, et al. Research on cloud detection method of domestic high-resolution satellite images[J]. Bulletin of Surveying and Mapping, 2020(11):66-70.
[12] 张家强, 李潇雁, 李丽圆, 等. 基于深度残差全卷积网络的Landsat8遥感影像云检测方法[J]. 激光与光电子学进展, 2020, 57(10):364-371.
[12] Zhang J Q, Li X Y, Li L Y, et al. Landsat8 remote sensing image based on deep residual fully convolutional network[J]. Laser and Optoelectronics Progress, 2020, 57(10):364-371.
[13] 张永宏, 蔡朋雁, 陶润喆, 等. 基于改进U-Net网络的遥感图像云检测[J]. 测绘通报, 2020(3):17-20.
[13] Zhang Y H, Cai P Y, Tao R Z, et al. Cloud detection for remote sensing images using improved U-Net[J]. Bulletin of Surveying and Mapping, 2020(3):17-20.
[14] 康超萌. 基于神经网络的国产高分光学遥感图像云检测[D]. 西安: 中国科学院大学(中国科学院西安光学精密机械研究所), 2018:20-27.
[14] Kang C M. Cloud detection in domestic high-resolution remote sensing image based deep neural networks[D]. Xi’an: University of Chinese Academy of Sciences, 2018:20-27.
[15] 张晨, 张秀再, 杨昌军. 双注意力RDA-Net遥感影像云与云阴影检测方法[J/OL]. 激光与光电子学进展, 2021:1-19. https://kns.cnki.net/kcms/detail/31.1690.TN.20210222.1114.023.html.
url: https://kns.cnki.net/kcms/detail/31.1690.TN.20210222.1114.023.html
[15] Zhang C, Zhang X Z, Yang C J. Dual attention RDA-Net for cloud and cloud shadow detection of remote sensing image[J/OL]. Laser and Optoelectronics Progress, 2021:1-19. https://kns.cnki.net/kcms/detail/31.1690.TN.20210222.1114.023.html.
url: https://kns.cnki.net/kcms/detail/31.1690.TN.20210222.1114.023.html
[16] Zhou L C, Zhang C, Wu M. D-LinkNet:LinkNet with pretrained encoder and dilated convolution for high resolution satellite imagery road extraction[C]// IEEE Conference on Computer Vision and Pattern Recognition (CVPR) Workshops, 2018:182-186.
[17] 彭博. 基于深度学习的遥感图像道路信息提取算法研究[D]. 成都: 电子科技大学, 2019:45-46.
[17] Peng B. Research on road information extraction of remote sensing image based on deep learning[D]. Chengdu: University of Electronic Science and Technology of China, 2019:45-46.
[18] Cao Y, Xu J R, Stephen L, et al. GCNet:Non-local networks meet squeeze-excitation networks and beyond[C]// IEEE CVF International Conference on Computer Vision (ICCV), 2019:1971-1980.
[19] Fu J, Liu J, Tian H, et al. Dual attention network for scene segmentation[C]// IEEE Conference on Computer Vision and Pattern Recognition, 2019:3146-3154.
[20] Huang G, Liu Z, Van Der Maaten L, et al. Densely connected convo-lutional networks[C]// Proceedings of the 30th IEEE Conference on Computer Vision and Pattern Recognition(CVPR),Honolulu, 2017.
[21] He K M, Zhan X Y, Ren S Q. Deep residual learning for image reco-gnition[C]// IEEE Conference on Computer Vision and Pattern Recognition, 2016:770-778.
[22] Hu J, Shen L, Sun G. Squeeze-and-excitation networks[C]// IEEE Conference on Computer Vision and Pattern Recognition(CVPR), 2018:7132-7141.
[23] Woo S, Park J, Lee J Y, et al. CBAM:Convolutional block attention module[C]// 15th European Conference on Computer Vision (ECCV), 2018, 11211:3-19.
[24] Wang X L, Girshick R, Gupta A, et al. Non-local neural networks[C]// IEEE Conference on Computer Vision and Pattern Recognition, 2018:7794-7803.
[25] Chen L C, Papandreou G, Kokkinos I, et al. DeepLab:Semantic image segmentation with deep convolutional nets,atrous convolution,and fully connected CRFs[J]. IEEE Transactions on Pattern Analysis and Machine Interlligence, 2018, 40(4):834-848.
[1] WU Weichao, YE Fawang. Cloud detection of Sentinel-2 images for multiple backgrounds[J]. Remote Sensing for Natural Resources, 2023, 35(3): 124-133.
[2] LI Guoxiang, XIA Guo’en, BAI Liming, MA Wenbin. Remote sensing image classification based on DenseNet feature hashing[J]. Remote Sensing for Natural Resources, 2023, 35(1): 66-73.
[3] YAN Junjie, GUO Xuexing, QU Jianhua, HAN Min. An FY-4A/AGRI cloud detection model based on the naive Bayes algorithm[J]. Remote Sensing for Natural Resources, 2022, 34(3): 33-42.
[4] LI Xusheng, LIU Yufeng, CHEN Donghua, LIU Saisai, LI Hu. Cloud detection based on support vector machine with image features for GF-1 data[J]. Remote Sensing for Land & Resources, 2020, 32(3): 55-62.
[5] Zhiwei LIN, Weihao TU, Jiahang HUANG, Qilu DING, Zhengwen ZHOU, Jinfu LIU. Tree species recognition of UAV aerial images based on FC-DenseNet[J]. Remote Sensing for Land & Resources, 2019, 31(3): 225-233.
[6] LIU Zihan, WU Yanlan. A review of cloud detection methods in remote sensing images[J]. REMOTE SENSING FOR LAND & RESOURCES, 2017, 29(4): 6-7.
[7] XIA Lang, MAO Kebiao, SUN Zhiwen, MA Ying, ZHAO Fen. Method for detecting cloud at night from VIIRS data based on DNB[J]. REMOTE SENSING FOR LAND & RESOURCES, 2014, 26(3): 74-79.
[8] QIN Yan, DENG Ru-ru, HE Ying-qing, CHEN Lei, CHEN Qi-dong, XIONG Shou-ping. Algorithm for Removing Thick Clouds in TM Image Based on Spectral and Geometric Information[J]. REMOTE SENSING FOR LAND & RESOURCES, 2012, 24(4): 55-61.
[9] HAN Jie, YANG Lei-ku, LI Hui-fang, LIANG Hong-you, MA Xiao-hong, XIE Yu-juan. Research on Algorithm of Cloud Detection for HJ-1B Image Based on Dynamical Thresholding[J]. REMOTE SENSING FOR LAND & RESOURCES, 2012, 24(2): 12-18.
[10] HOU Yue, LIU Pei-Xun, CHEN Shun-Yun, LIU Shao-Feng. A STUDY OF NIGHT CLOUD DETECTION BASED ON MODIS IMAGE[J]. REMOTE SENSING FOR LAND & RESOURCES, 2008, 20(1): 34-37.
[11] HE Quan-Jun, CAO Jing, HUANG Jiang, WU Zhi-Jun. CLOUD DETECTION IN MODIS DATA BASED ON MULTI-SPECTRUM SYNTHESIS[J]. REMOTE SENSING FOR LAND & RESOURCES, 2006, 18(3): 19-22.
[12] JIANG Geng-ming, NIU Zheng, RUAN Wei-li, WANG Chang-yao . CLOUD-FREE COMPOSITION OF MODIS DATA AND ALGORITHM REALIZATION[J]. REMOTE SENSING FOR LAND & RESOURCES, 2004, 16(2): 11-15,79.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
京ICP备05055290号-2
Copyright © 2017 Remote Sensing for Natural Resources
Support by Beijing Magtech