[发明专利]一种红外图像的转换训练方法、装置、设备及存储介质

摘要

本发明公开了一种红外图像的转换训练方法、装置、设备及计算机可读存储介质，该方法包括：获取第一图像域集和第二图像域集；基于帧间差一致性和对比学习，利用第一图像域集和第二图像域集对预设生成器和预设判别器的权重参数进行迭代训练，获取训练完成的预设生成器；根据训练完成的预设生成器，获取转换生成器；本发明引入对比学习的思想避免了现有循环一致性思想的严格的双向映射的方式，能够更好地适用存在时间跨度的红外图像转换，使得转换的白天真彩图像能够保留夜间红外图像的原始语义结构信息；基于帧间差一致性，采用帧间差分的思想约束输入与输出之间的帧间差异，能够有效防止生成真彩视频产生帧间闪烁问题。