[发明专利]舌诊图像的颜色校正模型训练方法、颜色校正方法及设备

权利要求书

1.一种舌诊图像的颜色校正模型训练方法，其特征在于，包括：

获取预设颜色的闪光状态下的拍摄的舌部图像，作为初始图像，得到若干所述初始图像，所述预设颜色为N种，N为大于3的正整数；

针对每张所述初始图像，采用语义分割网络，识别所述初始图像中的舌部区域，并对所述舌部区域生成第一标记，对舌部区域外的区域生成第二标记，得到包含第一标记和第二标记的训练图像；

根据所述训练图像和所述训练图像对应预设颜色，确定每种预设颜色对应的偏色信息；

基于所述偏色信息和所述训练图像进行初始颜色校正模型的训练，得到训练好的舌诊图像的颜色校正模型。

2.如权利要求1所述的舌诊图像的颜色校正模型训练方法，其特征在于，所述预设颜色包括白色、红色、绿色和蓝色，获取所述初始图像时的拍摄参数相同。

3.如权利要求2所述的舌诊图像的颜色校正模型训练方法，其特征在于，所述根据所述训练图像和所述训练图像对应预设颜色，确定每种预设颜色对应的偏色信息，包括：

在自然光状态下拍摄得到参考图像，所述参考图像的拍摄参数与所述初始图像的拍摄参数相同；

将所述训练图像中第一标记区域的图像，作为识别区域图像，并获取所述参考图像中与所述第一标记区域对应区域的图像，作为参考区域图像；

将所述识别区域图像与所述参考区域图像，确定所述训练图像对应预设颜色的偏色信息。

4.如权利要求2所述的舌诊图像的颜色校正模型训练方法，其特征在于，在所述预设颜色为白色时，所述获取预设颜色的闪光状态下的拍摄的舌部图像，作为初始图像，包括：在拍摄视野范围内，设置一白色参照物；

所述根据所述训练图像和所述训练图像对应预设颜色，确定每种预设颜色对应的偏色信息包括：

将所述训练图像中第一标记区域的图像，与所述白色参照物进行偏色计算，得到所述白色的偏色信息。

5.如权利要求1至4任一项所述的舌诊图像的颜色校正模型训练方法，其特征在于，所述初始颜色校正模型的训练为机器学习模型或者神经网络模型。

6.一种舌诊图像的颜色校正方法，其特征在于，包括：

获取待校准的舌诊图像；

采用训练好的舌诊图像的颜色校正模型对所述待校准的舌诊图像进行校准，得到校准图像，其中，所述训练好的舌诊图像的颜色校正模型根据权利要求1至5任一项所述的方法训练得到。

7.一种舌诊图像的颜色校正模型训练装置，其特征在于，所述舌诊图像的颜色校正模型训练装置包括：

初始图像获取模块，用于获取预设颜色的闪光状态下的拍摄的舌部图像，作为初始图像，得到若干所述初始图像，所述预设颜色为N种，N为大于3的正整数；

训练图像生成模块，用于针对每张所述初始图像，采用语义分割网络，识别所述初始图像中的舌部区域，并对所述舌部区域生成第一标记，对舌部区域外的区域生成第二标记，得到包含第一标记和第二标记的训练图像；

偏色信息确定模块，用于根据所述训练图像和所述训练图像对应预设颜色，确定每种预设颜色对应的偏色信息；

校正模型训练模块，用于基于所述偏色信息和所述训练图像进行初始颜色校正模型的训练，得到训练好的舌诊图像的颜色校正模型。

8.一种舌诊图像的颜色校正装置，其特征在于，包括：

待校准图像获取模块，用于获取待校准的舌诊图像；

图像校准模块，用于采用训练好的舌诊图像的颜色校正模型对所述待校准的舌诊图像进行校准，得到校准图像，其中，所述训练好的舌诊图像的颜色校正模型根据权利要求1至5任一项所述的方法训练得到。

9.一种计算机设备，包括存储器、处理器以及存储在所述存储器中并可在所述处理器上运行的计算机程序，其特征在于，所述处理器执行所述计算机程序时实现如权利要求1至5任一项所述的舌诊图像的颜色校正模型训练方法，或者，所述处理器执行所述计算机程序时实现如权利要求6所述的舌诊图像的颜色校正方法。

10.一种计算机可读存储介质，所述计算机可读存储介质存储有计算机程序，其特征在于，所述计算机程序被处理器执行时实现如权利要求1至5任一项所述的舌诊图像的颜色校正模型训练方法，或者，所述计算机程序被处理器执行时实现如权利要求6所述的舌诊图像的颜色校正方法。