激光与红外

作者：叶明亮

单位：1.沈阳理工大学机械工程学院,辽宁 沈阳 110159；2.沈阳理工大学装备工程学院,辽宁 沈阳 110159；

关键词：生成对抗网络;pix2pix;图像转换;残差结构

针对不同波段图像获取代价不同的问题,提出一种基于pix2pix的图像转换方法并进行改进。主要针对生成器和鉴别器两方面进行改进。生成器方面,使用残差结构的生成器替换原来的U-Net生成器以缓解梯度消失问题；引入可变形卷积,提高目标边缘和小目标的生成效果；引入BAM注意力机制,提高了算法对图像中主要目标的特征提取能力以提升生成图像的效果。鉴别器方面:改变PatchGAN中卷积层的层数(原PatchGAN为3层卷积),设置对照实验找到转换效果最好的卷积层数。以可见光图像和红外图像之间的转换为例进行实验。实验结果表明,改进后的算法在生成图像上的均方根误差(MSE)下降了314%、结构相似性(SSIM)提高了112%,可以更好的实现红外图像和可见光图像之间的转换。

来源：2024年第7期

《激光与红外》期刊编辑部

查看激光与红外杂志2024年第7期