[发明专利]内镜图像颜色还原方法及装置

说明书

本发明提供一种内镜图像颜色还原方法及装置，方法包括：将待还原内镜图像转换为灰度图像；基于预先训练好的下采样卷积神经网络提取所述灰度图像的局部特征，基于预先训练好的ResNet网络提取所述灰度图像的全局场景信息，将所述局部特征和所述全局场景信息进行融合；基于预先训练好的上采样卷积神经网络将所述融合结果恢复到所述待还原内镜图像的尺寸，并根据尺寸还原后的所述融合结果基于所述上采样卷积神经网络获取所述灰度图像对应的色度信息，根据所述灰度图像对应的色度信息对所述待还原内镜图像进行颜色还原。本发明能实现对失真或偏色图像进行颜色还原。

技术领域

本发明属于图像处理技术领域，尤其涉及一种内镜图像颜色还原方法及装置。

背景技术

内窥镜手术过程中，由于照明结构的空间限制、多种照明光源的干扰和光源输出功率过高等原因，会造成内窥镜图像产生明暗不均、高曝光、或者严重偏色失真的问题，导致图像失去大量的色度信息和细节信息。此时医生无法实时判断内窥镜镜头在腔道内的位置和病灶反应，增加了手术风险。

因此，亟需一种内镜图像颜色还原方法，能还原内窥镜图像的颜色，从而还原手术的真实场景，确保医生手术的精准度和安全性。

发明内容

为克服上述内窥镜图像出现偏色或色度信息丢失，无法还原手术的真实场景，影响医生手术的精确度和安全性的问题或者至少部分地解决上述问题，本发明实施例提供一种内镜图像颜色还原方法及装置。

根据本发明实施例的第一方面，提供一种内镜图像颜色还原方法，包括：

将待还原内镜图像转换为灰度图像；

基于预先训练好的下采样卷积神经网络提取所述灰度图像的局部特征，基于预先训练好的ResNet网络提取所述灰度图像的全局场景信息，将所述局部特征和所述全局场景信息进行融合；其中，所述下采样卷积神经网络为包含多个下采样层的卷积神经网络；

基于预先训练好的上采样卷积神经网络将所述融合结果恢复到所述待还原内镜图像的尺寸，并根据尺寸还原后的所述融合结果基于所述上采样卷积神经网络获取所述灰度图像对应的色度信息，根据所述灰度图像对应的色度信息对所述待还原内镜图像进行颜色还原；其中，所述上采样卷积神经网络为包含多个上采样层的卷积神经网络。

根据本发明实施例第二方面提供一种内镜图像颜色还原装置，包括：

转换模块，用于将待还原内镜图像转换为灰度图像；

提取模块，用于基于预先训练好的下采样卷积神经网络提取所述灰度图像的局部特征，基于预先训练好的ResNet网络提取所述灰度图像的全局场景信息，将所述局部特征和所述全局场景信息进行融合；其中，所述下采样卷积神经网络为包含多个下采样层的卷积神经网络；

还原模块，用于基于预先训练好的上采样卷积神经网络将所述融合结果恢复到所述待还原内镜图像的尺寸，并根据尺寸还原后的所述融合结果基于所述上采样卷积神经网络获取所述灰度图像对应的色度信息，根据所述灰度图像对应的色度信息对所述待还原内镜图像进行颜色还原；其中，所述上采样卷积神经网络为包含多个上采样层的卷积神经网络。

根据本发明实施例的第三个方面，还提供一种电子设备，包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序，所述处理器调用所述程序指令能够执行第一方面的各种可能的实现方式中任一种可能的实现方式所提供的内镜图像颜色还原方法。

根据本发明实施例的第四个方面，还提供一种非暂态计算机可读存储介质，所述非暂态计算机可读存储介质存储计算机指令，所述计算机指令使所述计算机执行第一方面的各种可能的实现方式中任一种可能的实现方式所提供的内镜图像颜色还原方法。