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

说明书

本发明公开了一种舌诊图像的颜色校正模型训练方法、颜色校正方法及设备，包括：获取预设颜色的闪光状态下的拍摄的舌部图像，作为初始图像，得到若干初始图像，预设颜色为N种，N为大于3的正整数；针对每张初始图像，采用语义分割网络，识别初始图像中的舌部区域，并对舌部区域生成第一标记，对舌部区域外的区域生成第二标记，得到包含第一标记和第二标记的训练图像；根据训练图像和训练图像对应预设颜色，确定每种预设颜色对应的偏色信息；基于偏色信息和训练图像进行初始颜色校正模型的训练，得到训练好的舌诊图像的颜色校正模型。使得得到的颜色校正模型可以快速识别出不同光线对真实舌部的颜色造成的偏色，有利于提高颜色校正的准确性。

技术领域

本发明涉及数据处理领域，尤其涉及一种舌诊图像的颜色校正模型训练方法、颜色校正方法及设备。

背景技术

随着电子信息技术的发展，线上问诊越来越方便人们的生活，舌诊是在中医诊断中应用广泛，并得到多数人的认可。目前已经开发出专业的舌诊仪器，用户只需要在一个专业的机器上伸出舌头拍照，就可以得到专业的检测报告。也有一些公司，提供舌诊的app，远程帮助用户做舌诊检测。专业的舌诊仪器虽然精准但价格昂贵，而且必须挂号到医院去做；对于一些亚健康的小毛病，没必要往医院跑。

舌诊仪器会发出标准的白光照射舌头然后拍照诊断，能精准的知道舌头的颜色；但舌诊app是借助自然光线，不同光照情况下拍的照片对舌头颜色的判断影响很大，可能在夕阳下，可能在日光灯下，也可能在白天的房间里，自然光的情况多种多样，所以很多舌诊app换个环境下拍摄就会得到不同的诊断结果，人们使用舌诊app拍摄舌头照片，可能在夕阳下，可能在日光灯下，也可能在白天的房间里，自然光的情况多种多样，没办法控制，这使得拍摄的图像颜色存在误差，舌诊图像质量达不到应用的要求。

发明内容

本发明实施例提供一种舌诊图像的颜色校正模型训练方法、颜色校正方法、装置、计算机设备和存储介质，以提高舌诊图像的质量。

为了解决上述技术问题，本申请实施例提供一种舌诊图像的颜色校正模型训练方法，所述舌诊图像的颜色校正模型训练方法包括：

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

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

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

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

可选地，所述预设颜色包括白色、红色、绿色和蓝色，获取所述初始图像时的拍摄参数相同。

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

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

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

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

可选地，在所述预设颜色为白色时，所述获取预设颜色的闪光状态下的拍摄的舌部图像，作为初始图像，包括：在拍摄视野范围内，设置一白色参照物；

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

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