一种基于拓扑语义结构损失的无人机红外图像着色方法

本发明公开了一种基于拓扑语义结构损失的无人机红外图像着色方法，该方法包含拓扑结构语义约束模块和判别器引导的采样策略。红外图像和彩色图像虽然在光谱上不同，但它们表达相同的场景结构。所以，拓扑结构语义约束模块共享红外和彩色图像特定补丁之间的邻接矩阵，利用图神经网络捕捉不同域图结构的节点特征表示，并通过对比损失提升图结构节点间的互信息，确保拓扑结构的语义一致性，从而保留场景的结构信息。判别器引导注意力采样策略在训练过程能够选择最具信息量的补丁，既引导生成器在编解码阶段保持内容一致性，又引导拓扑感知语义一致性。基于KAIST数据集和FLIR数据集的实验证明本发明方法在着色性能上优于最先进的方法。