[发明专利]一种图像着色方法、装置、系统及存储介质

说明书

本发明提供了一种图像着色方法、装置、系统及存储介质，所述方法包括：将待着色图像转换为Lab颜色空间格式得到所述待着色图像的L通道值；获取用于着色的图像或所述待着色图像的ab通道值，其中，所述用于着色的图像包括与所述待着色图像至少部分相同的图像；将所述L通道值和所述ab通道值输入至图像着色神经网络，得到用于着色的ab通道值；将所述用于着色的ab通道值与所述L通道值融合得到所述待着色图像的图像着色结果。根据本发明的方法、装置、系统及存储介质，通过神经网络得到着色的颜色通道值与待着色图像的亮度通道融合实现图像着色，改善现有的图像着色偏色、漏色、缺色等问题，提高了图像着色的效率以及着色效果。

技术领域

本发明涉及图像处理技术领域，更具体地涉及图像着色的处理。

背景技术

图像着色旨在对一张已有图像填充颜色使其成为一张不出现肉眼可见的明显缺陷的彩色图片。例如，基于少量的原始颜色信息的着色是指，在Lab颜色空间中，一张图片中大部分像素没有ab的颜色信息，少量像素有ab颜色信息，需要根据已有少量颜色对没有颜色的像素进行着色，即恢复其UV通道。现有的着色方法是基于像素灰度值和两个像素之间的距离等信息，并根据假设：“对于相邻的两个像素，如果其亮度比较相似，那么其颜色也应该相似”设计目标函数并求解稀疏线性方程组，从而实现根据假设以及先验信息推测出某个像素点的颜色。这种方法存在一定缺陷，因为该方法是基于假设，该假设只是一种经验性假设，并不完全可靠，容易出现图像某些区域漏色，缺色以及偏色等问题。

此外，现有的着色方法中如果颜色信息ab并不能和原始图像的L通道像素级别严格对齐，很容易导致将颜色填充到错误的位置，也无法发掘颜色和物体轮廓信息之间的关系从而将正确的颜色填充到正确的物理轮廓中。例如相机两个摄像头拍摄出的两张图像，第一张拥有ab通道颜色信息，第二张只有L通道信息，需要利用第一张中的颜色信息为第二张进行着色，因为两个摄像头不在同一位置并且参数不同，使得两张图片存在景深和偏移等不同，使用现有的着色技术将会导致图像着色出现错误，使得图像着色效果不好，降低了用户体验。

因此，现有技术中的图像着色存在着色效果不好的问题。

发明内容

考虑到上述问题而提出了本发明。本发明提供了一种图像着色方法、装置、系统及计算机存储介质，通过神经网络得到着色的颜色通道值与待着色图像的亮度通道融合实现图像着色，改善现有的图像着色偏色、漏色、缺色等问题，提高了图像着色的效率以及着色效果。

根据本发明的第一方面，提供了一种图像着色方法，包括：

将待着色图像转换为Lab颜色空间格式得到所述待着色图像的L通道值；

获取用于着色的图像或所述待着色图像的ab通道值，所述用于着色的图像包括与所述待着色图像至少部分相同的图像；

将所述L通道值和所述ab通道值输入至图像着色神经网络，得到用于着色的ab通道值；

将所述用于着色的ab通道值与所述L通道值融合得到所述待着色图像的图像着色结果。

根据本发明的第二方面，提供了一种图像着色装置，包括：

待着色图像模块，用于将待着色图像转换为Lab颜色空间格式得到所述待着色图像的L通道值；

着色图像模块，用于获取用于着色的图像或所述待着色图像的ab通道值，所述用于着色的图像包括与所述待着色图像至少部分相同的图像；

计算模块，用于将所述L通道值和所述ab通道值输入至图像着色神经网络，得到用于着色的ab通道值；

着色模块，用于将所述用于着色的ab通道值与所述L通道值融合得到所述待着色图像的图像着色结果。