A method for color consistency of remote sensing images based on generative adversarial networks
-
摘要: 在遥感成像过程中易在拍摄影像内部、影像与影像之间产生亮度不均匀、色彩不一致的现象,通过人工借助图像处理软件进行色彩调节已经不能满足呈几何级数量增长的遥感影像调色需求,因此提出一种针对土地利用率高的复杂城区地物的融合注意力机制无监督循环一致生成对抗网络(channel attention-cycle generative adversarial networks,CA-CycleGAN)。首先,通过直方图调整和Photoshop等软件手工制作用于色彩参考的样本数据集,选择合适的城区影像数据作为待校正影像样本集,将2部分影像分别进行裁切,得到预处理后的影像样本集; 然后,将处理好的待校正影像集和色彩参考影像集通过CA-CycleGAN中,由于在生成器中加入了注意力机制,因此在生成器与鉴别器相互对抗的训练过程中能够利用注意力特征图将生成的重点分配在重要的区域,提高生成影像效果,得到基于城区影像的色彩校正模型以及色彩校正后的影像图。影像校正效果和损失函数图表明,所提出的方法在循环一致生成对抗网络基础上做出了优化,加入注意力机制的生成对抗网络在调整影像色彩上的综合表现效果优于不加注意力机制的生成对抗网络。相较于传统方法大大减少了色彩校正的时间,对比人工调色增加了影像色彩校正效果的稳定性。证明所提出方法在遥感影像匀色工作中优势较明显,具有较好的应用前景。Abstract: Uneven brightness and inconsistent colors are prone to occur inside and between captured images in the process of remote sensing imaging. However, the manual color conditioning combined with image processing software can no longer meet the color matching demand of geometrically increasing remote sensing images. Given this, this study proposed a kind of unsupervised channel-cycle generative adversarial network (CA-CycleGAN) integrated with the attention mechanism suitable for ground objects in complex urban areas with a high land utilization rate. Firstly, the sample data set used for color reference was manually made through histogram adjustment and Photoshop, and the appropriate urban images were selected as the sample set to be corrected. Then, the two kinds of images were cut respectively to obtain the preprocessed image sample sets. Finally, the preprocessed image set to be corrected and the image set for color reference were processed using the CA-CycleGAN. Because the attention mechanism has been added to the generator, the generated focuses can be distributed into key areas using the attention feature map in the training process of the confrontation between the generator and the discriminator, thus improving the image effects and obtaining the color correction model based on urban images and the images after color correction. Both the image correction effect and the loss function diagram show that the proposed method is optimized based on the CycleGAN and that the comprehensive performance of the CycleGAN integrated with the attention mechanism is better than that without the attention mechanism. Compared to conventional methods, the method proposed in this study greatly reduced the time for color correction and achieved more stable image color correction effects than manual color matching. Therefore, the method proposed in this study enjoys significant advantages in the color dodging of remote sensing images and has a good application prospect.
-
Key words:
- remote sensing /
- image color correction /
- CycleGAN /
- urban satellite image /
- attention mechanism
-
-
[1] 金淑英, 包更生, 马洪超, 等. 海港海底视频影像的灰度不均匀校正方法的研究[J]. 武汉测绘科技大学学报, 2000, 25(6):531-536.
[2] Jin S Y, Bao G S, Ma H C, et al. Research on gray nonuniformity correction method of seaport seafloor video image[J]. Geomatics and Information Science of Wuhan University, 2000, 25(6):531-536.
[3] Kim S J, Kim N Y, Lee B G, et al. Wavelet based uneven illumination compensation for defect detection of flat panel display[C]// Iasted International Conference on Signal Processing,Pattern Reco-gnition,and Applications.ACTA Press, 2007:104-108.
[4] Tan R T. Visibility in bad weather from a single image[C]// IEEE Conference on Computer Vision and Pattern Recognition,IEEE, 2008:1-8.
[5] 谭攀, 吴绍斐, 苏晓军, 等. 基于影像动态分块与HIS结合的匀光算法[J]. 影像技术, 2014, 26(1):55-56.
[6] Tan P, Wu S F, Su X J, et al. A smoothing algorithm based on dynamic image segmentation and HIS[J]. Image Technology, 2014, 26(1):55-56.
[7] 张振. 光学遥感影像匀光算法研究[D]. 郑州: 解放军信息工程大学, 2010.
[8] Zhang Z. Research on dodging methods of optical remotely sensed image[D]. Zhengzhou: Information Engineering University, 2010.
[9] 易磊. 遥感影像色彩一致性处理技术研究[D]. 郑州: 解放军信息工程大学, 2015.
[10] Yi L. Research on color consistency processing for remote sensing image[D]. Zhengzhou: Information Engineering University, 2015.
[11] 任中杰. 遥感瓦片影像色彩一致性研究[D]. 赣州: 江西理工大学, 2019.
[12] Ren Z J. Research on color consistency of remote sensing tile image[D]. Ganzhou: Jiangxi University of Science and Technology, 2019.
[13] 杜神斌. 数字正射影像镶嵌中色彩一致性方法研究[D]. 南昌: 东华理工大学, 2018.
[14] Du S B. Research on color consistency method in digital orthophoto mosaic[D]. Nanchang: East China University of Technology, 2018.
[15] 刘晓龙. 基于影像匹配接边纠正的数字正射影像的镶嵌技术[J]. 遥感学报, 2001, 5(2):104-109.
[16] Liu X L. Image match and overlapping area correction mosaic technology for digital orthophoto maps[J]. National Remote Sensing Bulletin, 2001, 5(2):104-109.
[17] 王密, 潘俊. 一种数字航空影像的匀光方法[J]. 中国图象图形学报, 2004(6):104-108,127.
[18] Wang M, Pan J. A method of removing the uneven illumination for digital aerial image[J]. Journal of Image and Graphics, 2004(6):104-108,127.
[19] 刘建涛. 基于彩色空间变换的航空数码影像自动匀光处理研究[D]. 西安: 长安大学, 2008.
[20] Liu T J. Study on dodging of aerial digital image based on color model transform[D]. Xi’an: Chang’an University, 2008.
[21] 李治江. 彩色影像色调重建的理论与实践[D]. 武汉: 武汉大学, 2005.
[22] Li Z J. Theory and practice on tone reproduction of color photos[D]. Wuhan: Wuhan University, 2005.
[23] Stumpfel J, Jones A, Wenger A, et al. Direct HDR capture of the sun and sky[C]// Proceedings of the 3rd International Conference on Computer Graphics,Virtual Reality,Visualization and Interaction in Africa, 2004:145-149.
[24] 张振, 朱宝山, 朱述龙. 反差一致性改进的MASK匀光算法[J]. 测绘科学技术学报, 2010(1):54-56.
[25] Zhang Z, Zhu B S, Zhu S L. An improved MASK dodging method based on contrast consistency processing[J]. Journal of Geomatics Science and Technology, 2010(1):54-56.
[26] 朱述龙, 朱宝山, 王红卫. 遥感影像处理与应用[M]. 北京: 科学出版社, 2006.
[27] Zhu S L, Zhu B S, Wang H W. Remote sensing image processing and application[M]. Beijing: Science Press, 2006.
[28] Orsini G, Ramponi G, Carrai P, et al. A modified retinex for image contrast enhancement and dynamics control[C]// International Conference on Image Processing.IEEE, 2003:III-393-396.
[29] Lam E Y. Combining gray world and retinex theory for automatic white balance in digital photography[C]// International Symposium on Consumer Electronics.IEEE, 2005:134-139.
[30] Nnolim U, Lee P. Homomorphic filtering of colour images using a spatial filter kernel in the HIS colour space[J]. 2008, 71(2):1738-1743.
[31] Seow M J, Asari V K. Ratio rule and homomorphic filter for enhancement of digital colour image[J]. Neurocomputing, 2006, 69(7-9):954-958.
[32] Xue Z X. Semi-supervised convolutional generative adversarial network for hyperspectral image classification[J]. IET Image Processing, 2020, 14(4):709-719.
[33] 李雪, 张力, 王庆栋, 等. 多时相遥感影像语义分割色彩一致性对抗网络[J]. 测绘学报, 2020, 49(11):1473-1484.
[34] Li X, Zhang L, Wang Q D, et al. Multi-temporal remote sensing imagery semantic segmentation color consistency adversarial network[J]. Acta Geodaetica Cartographica Sinica, 2020, 49(11):1473-1484.
[35] 王照乾, 孔韦韦, 滕金保, 等. DenseNet生成对抗网络低照度图像增强方法[J]. 计算机工程与应用, 2020, 58(8):214-220.
[36] Wang Z Q, Kong W W, Teng J B, et al. Low illumination image enhancement method based on DenseNet GAN[J]. Computer Engineering and Applications, 2020, 58(8):214-220.
[37] Lyu N, Ma H X, Chen C, et al. Remote sensing data augmentation through adversarial training[J]. IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing, 2020, 14:9318-9333.
[38] 何鎏一. 基于深度学习的去雾算法设计[D]. 青岛: 青岛大学, 2020.
[39] He L Y. Design of defogging algorithm based on deep learning[D]. Qingdao: Qingdao University, 2020.
[40] Hu A, Xie Z, Xu Y, et al. Unsupervised haze removal for high-resolution optical remote-sensing images based on improved generative adversarial networks[J]. Remote Sensing, 2020, 12(24):4162.
[41] 冀俭俭. 基于生成对抗网络的分级联合图像补全方法[D]. 北京: 北京林业大学, 2020.
[42] Ji J J. Hierarchical jiont image completion method based on generative adversarial network[D]. Beijing: Beijing Forestry University, 2020.
[43] 郑凯, 李建胜, 王俊强, 等. DCLS-GAN:利用生成对抗网络的天绘一号卫星高原地区影像去云方法[J]. 测绘学报, 2021, 50(2):248-259.
[44] Zheng K, Li J S, Wang J Q, et al. DCLS-GAN:Cloud removal method for plateau area of TH-1 satellite image[J]. Acta Geodaetica Cartographica Sinica, 2021, 50(2):248-259.
[45] Zhu J Y, Park T, Isola P, et al. Unpaired image-to-image translation using cycle-consistent adversarial networks[C]// 2017 IEEE International Conference on Computer Vision.IEEE, 2017,17453078.
-
计量
- 文章访问数: 743
- PDF下载数: 33
- 施引文献: 0
下载: