[发明专利]夜间图像着色方法、装置、介质和设备

说明书

本发明公开了一种夜间图像着色方法、装置、介质和设备，包括：构建着色网络，并且对着色网络进行训练，得到着色模型；获取夜间同一视角拍摄的可见光图像和红外图像；根据预设融合规则，针对夜间可见光图像和红外图像进行图像融合处理，得到待着色的融合图像；将待着色的融合图像输入到着色模型中，进行特征提取和无参考着色。本发明通过将可见光图像和红外图像进行特征融合，综合可见光图像中的场景细节信息和红外图像中的目标特征信息，并且基于训练后的着色模型，实现无参考夜间图像的着色的方法，可以有效地提升夜间图像的着色质量。

技术领域

本发明涉及图像处理技术领域，特别涉及一种夜间图像着色方法、装置、介质和设备。

背景技术

夜间成像技术是指在夜间低照明度的条件下，通过利用光电探测或成像设备，将人眼不可视的场景转化为可视图像，从而扩展人眼在夜间低照度下的视力观察范围，实现夜间低照度下隐蔽信息的采集、处理和显示。

夜视成像技术在生产生活、交通、军事等领域有着重要的应用。例如，在交通出行领域，夜间技术的使用可以有效地扩大人眼的观察范围，进一步增强对目标的识别能力，极大地避免在低照度环境下交通事故的发生；而在军事领域，夜间技术可以让士兵在夜间克服复杂地形等环境障碍，提高夜间侦察、瞄准、射击等能力，强化军事技能。

然而，传统的微光夜视仪和红外成像仪等成像设备所生成的图像均为单色图像，缺乏色彩信息，难以观察目标的细节，使得观察者分辨不出目标之间的颜色差别，限制了对目标的识别和判断的能力。因此，通过利用计算机视觉技术，实现对夜间图像的着色，可以有效提升人眼对低照度场景的感知能力，增强对场景信息的认识和判断，有着重要的意义。

现有的夜间图像着色方法通常是利用一些参考图像着色方法进行着色。这种方法的问题在于参考图像的选取需要一定的工作量，好的参考图像可以实现较为准确的着色，而不好的参考图像则容易导致跟现实不符的着色效果，具有比较大的主观性。

发明内容

本发明的第一目的在于克服现有技术的缺点与不足，提供一种夜间图像着色方法，该方法通过将可见光图像和红外图像进行特征融合，实现无参考夜间图像的着色的方法，可以有效地提升夜间图像的着色质量。

本发明的第二目的在于提供一种夜间图像着色装置。

本发明的第三目的在于提供一种存储介质。

本发明的第四目的在于提供一种计算设备。

本发明的第一目的通过下述技术方案实现：一种夜间图像着色方法，步骤包括：

构建着色网络，并且对着色网络进行训练，得到着色模型；

获取夜间同一视角拍摄的可见光图像和红外图像；

根据预设融合规则，针对夜间可见光图像和红外图像进行图像融合处理，得到待着色的融合图像；

将待着色的融合图像输入到着色模型中，进行特征提取和无参考着色。

优选的，所述着色网络包括编码器、解码器、全连接层和softmax层，所述编码器用于从融合图像中提取特征；所述解码器用于从编码器所提取的特征中获取到颜色分布；

所述编码器包括第一嵌入模块和第一自注意力模块；

所述第一嵌入模块输入融合图像，对融合图像进行嵌入操作处理；第一嵌入模块嵌入操作后的图像输入到第一自注意力模块，由第一自注意力模块进行特征提取，提取的特征作为编码器的输出；

所述解码器包括第二嵌入模块和第二自注意力模块；

所述第二嵌入模块作为解码器的输入，输入图像进行嵌入操作处理；第二嵌入模块嵌入操作后的图像输入到第二自注意力模块；