[发明专利]图像生成装置获取方法及图像生成装置

说明书

本发明公开了一种图像生成装置获取方法及图像生成装置，第一图像生成装置用于将输入的连续至少两帧第一色度图像依次生成为对应的第二色度图像，以同步获取的连续帧第一色度图像和连续帧第二色度图像为样本进行图像生成装置、预测装置和判别装置互相博弈的训练过程，通过训练能够使图像生成装置生成图像更接近真实图像，预测装置生成的预测图像更接近真实图像。判别装置评价图像真实度的能力更强，本发明能够用于获得将灰度图像生成为彩色图像的图像生成装置，能够使生成的彩色图像更接近真实彩色图像，较好地保留颜色信息。

技术领域

本发明涉及图像处理技术领域，特别是涉及一种图像生成装置获取方法。本发明还涉及一种图像生成装置。

背景技术

在一些图像处理及应用中，依赖于能够获得高质量图像的前提。当环境亮度低或者夜间成像困难时，可见光相机无法获得高质量的图像或者视频，这对后续的图像应用或者图像处理带来了阻碍。

现有技术中，应用微光夜视技术或者红外夜视技术。微光夜视技术能够将夜间的微弱可见光通过增强几十万倍，来生成可见光图像，以此来实现夜间成像。红外夜视技术通过捕获红外辐射能够在极其苛刻的条件下获得红外光图像。由于人眼视觉神经系统只对可见光所成的真彩图像具有较高的分辨性，而微光夜视图像或者红外夜视图像，相比存在细节丢失以及为灰度图像，并不符合人的视觉感知习惯，也会导致各类针对彩色图像的处理算法性能显著下降。

发明内容

本发明的目的是提供一种图像生成装置获取方法，通过该方法能够获得将灰度图像生成为彩色图像的图像生成装置，能够使彩色图像较好地保留颜色信息。本发明还提供一种图像生成装置。

为实现上述目的，本发明提供如下技术方案：

一种图像生成装置获取方法，包括第一图像生成装置、第一预测装置、第一判别装置和第二判别装置，所述第一图像生成装置用于将输入的连续至少两帧第一色度图像依次生成为对应的第二色度图像，所述第一预测装置用于根据输入的连续至少两帧第一色度图像生成后一帧第一色度图像的预测图像，所述第一判别装置用于评价输入的第一色度图像的真实度，所述第二判别装置用于评价输入的第二色度图像的真实度；

所述方法包括：以连续至少两帧真实第一色度图像以及连续至少两帧真实第二色度图像为一组训练样本，该各帧真实第一色度图像与该各帧真实第二色度图像为对同一场景同步获取的图像，使用训练样本根据以下过程训练获得所述第一图像生成装置、所述第一预测装置、所述第一判别装置和所述第二判别装置，训练过程包括：

将一组训练样本的前至少一帧真实第一色度图像依次输入所述第一图像生成装置，通过所述第二判别装置分别评价生成的前至少一帧第二色度图像以及前至少一帧真实第二色度图像；

将一组训练样本的前至少一帧真实第一色度图像输入所述第一预测装置，获取生成的后一帧第一色度图像预测图像相对于后一帧真实第一色度图像的特征损失，通过所述第一判别装置分别评价生成的后一帧第一色度图像预测图像以及后一帧真实第一色度图像；

根据评价得出的各帧图像真实度以及获得的特征损失，训练所述第一图像生成装置、所述第一预测装置、所述第一判别装置和所述第二判别装置。

优选的，训练过程还包括：将一组训练样本的前至少一帧真实第二色度图像输入所述第一图像生成装置，分别获取相应生成的前至少一帧第二色度图像相对于对应的真实第二色度图像的特征损失。

优选的，训练过程还包括：将后一帧第一色度图像预测图像输入所述第一图像生成装置，通过所述第二判别装置评价后一帧真实第二色度图像以及由所述第一图像生成装置生成的后一帧第一色度图像预测图像对应的后一帧第二色度图像。

优选的，还包括第二图像生成装置，所述第二图像生成装置用于将输入的连续至少两帧第二色度图像依次生成为对应的第一色度图像，训练过程还包括：