[发明专利]图像着色方法、电子设备、存储介质及计算机程序产品

权利要求书

1.一种图像着色方法，其特征在于，所述图像着色方法包括：

获取待着色的灰度图像；

基于第一神经网络模型的至少两个分支，分别对所述灰度图像进行特征提取，得到至少两个特征图；其中，所述至少两个分支中不同分支对应的感受野不同，所述至少两个特征图与所述至少两个分支一一对应；

将所述至少两个特征图进行通道拼接，基于拼接后的特征图得到与所述灰度图像的尺寸相一致的二通道彩色图像；

将所述二通道彩色图像与所述灰度图像进行通道拼接，得到对所述灰度图像进行着色的第一彩色图像。

2.根据权利要求1所述的图像着色方法，其特征在于，所述方法还包括：

获取预先配置的多个不同三维显示查找表；其中，所述多个不同三维显示查找表至少包括三基色颜色通道中各颜色通道分别对应的三维显示查找表；

基于所述多个不同三维显示查找表，对所述第一彩色图像进行色彩调节，得到第二彩色图像。

3.根据权利要求2所述的图像着色方法，其特征在于，所述基于所述多个不同三维显示查找表，对所述第一彩色图像进行色彩调节，得到第二彩色图像，包括：

基于所述多个不同三维显示查找表，分别对所述第一彩色图像进行着色处理，得到多个不同着色图像；

基于所述灰度图像，确定对所述多个不同着色图像进行加权处理的第一权重，并以所述第一权重对所述多个不同着色图像进行加权处理，得到所述第二彩色图像。

4.根据权利要求2所述的图像着色方法，其特征在于，所述基于所述多个不同三维显示查找表，对所述第一彩色图像进行色彩调节，得到第二彩色图像，包括：

基于所述多个不同三维显示查找表，分别对所述第一彩色图像进行着色处理，得到多个不同着色图像；

基于所述灰度图像，确定对所述多个不同着色图像以及所述第一彩色图像进行加权处理的第二权重，并以所述第二权重对所述多个不同着色图像以及所述第一彩色图像进行加权处理，得到所述第二彩色图像。

5.根据权利要求3或4所述的图像着色方法，其特征在于，采用如下方式确定目标权重，所述目标权重包括第一权重或第二权重：

调用预先训练的第二神经网络模型，所述第二神经网络模型的输入为灰度图像，输出为目标权重；

基于所述第二神经网络模型的输出结果确定所述目标权重。

6.根据权利要求1至5中任意一项所述的图像着色方法，其特征在于，所述第一神经网络模型的至少两个分支，至少包括第一分支和第二分支；

其中，所述第一分支的感受野大于所述第二分支的感受野，且所述第一分支的感受野与所述第二分支的感受野之间的差异度大于目标差异度。

7.根据权利要求1至6中任意一项所述的图像着色方法，其特征在于，所述基于拼接后的特征图得到与所述灰度图像的尺寸相一致的二通道彩色图像，包括：

对所述拼接后的特征图进行通道间信息融合和/或空间信息融合，并对通道间信息融合和/或空间信息融合后的特征图进行尺寸调整及通道数调整，得到与所述灰度图像的尺寸相一致的二通道彩色图像。

8.一种电子设备，其特征在于，包括：

处理器；

用于存储处理器可执行指令的存储器；

其中，所述处理器被配置为：执行权利要求1至7中任意一项所述的图像着色方法。

9.一种存储介质，其特征在于，所述存储介质中存储有指令，当所述存储介质中的指令由处理器执行时，使得处理器能够执行权利要求1至7中任意一项所述的图像着色方法。

10.一种计算机程序产品，其特征在于，所述计算机程序产品包括计算机程序，所述计算机程序被处理器执行时实现权利要求1至7中任意一项所述的图像着色方法。