高级检索

基于深度神经网络的遥感图像彩色化方法

Remote Sensing Image Colorization Based on Deep Neural Networks with Multi-Scale Residual Receptive Filed

  • 摘要: 为了实现遥感图像彩色化,解决目前彩色化模型存在颜色不准确和颜色溢出等问题,提出一种端到端的深度神经网络模型.首先,通过构建多尺度残差感受域块提取丰富的高维特征;其次,利用U-Net、复杂残差结构、注意力机制和子像素向上卷积等结构构建一个彩色信息重建网络输出彩色化结果;最后,使用NWPU-RESISC45遥感图像数据集进行训练和验证.结果表明,与其他自动彩色化方法相比,所提方法的PSNR值平均提高6~10 dB,SSIM值增加0.05~0.11,实现了遥感图像彩色化.此外,该方法在RSSCN7和AID数据集上也取得了良好的彩色化效果.

     

    Abstract: To solve the problems of mistaken coloring and color bleeding in the current colorization methods,an end-to-end deep neural network is proposed to achieve remote sensing image colorization.First,the mul-ti-scale residual receptive filed net is introduced to extract the key features of source image.Second,a color information recovery network is con-structed by using U-Net,complex residual structure,attention mecha-nism,sequeeze-and-excitation and pixel-shuffle blocks to obtain color result.NWPU-RESISC45 dataset is chosen for model training and validation.Compared with other color methods,the PSNR value of the pro-posed method is increased by 6-10 dB on average and the SSIM value is increased by 0.05-0.11.In addition,the proposed method also achieves satisfactory color results on RSSCN7 and AID datasets.

     

/

返回文章
返回