图像处理模型训练方法及相关装置、设备和存储介质制造方法及图纸

本申请公开了一种图像处理模型训练方法及相关装置、设备和存储介质，方法包括：获取样本图像集；其中，样本图像集包括第一样本图像和样本相关图像，第一样本图像标注有目标对象的样本轮廓，样本相关图像的像素点的像素值表示包括第一样本图像中的第一像素点与样本轮廓的样本相关度；利用第一处理模型对第一样本图像进行预测，得到预测图像集；其中，预测图像集包括预测相关图像，预测相关图像包括第一像素点经预测与样本轮廓的预测相关度；基于样本相关图像与预测相关图像，获取第一差异；基于第一差异，调整第一处理模型的网络参数。通过该方法，实现了利用弱标签数据对第一处理模型的训练。的训练。的训练。

【技术实现步骤摘要】
图像处理模型训练方法及相关装置、设备和存储介质


[0001]本申请涉及深度学习
，特别是涉及一种图像处理模型训练方法及相关装置、设备和存储介质。

技术介绍

[0002]深度学习的快速发展，神经网络算法的日益改进，计算准确度也相应提高。目前，各行各业使用神经网络模型进行工作已经成为常态。例如，在医学领域，利用神经网络模型进行医学图像分割。又如，在交通领域，利用神经网络模型进行车辆识别。
[0003]目前，需要使用相当多的像素级的稠密标注数据来对模型进行训练才能获得比较的效果。但是，像素级的稠密标注数据获得具有一定的难度，成本较高，这也使得训练神经网络的成本较高，限制了神经网络的进一步普及。
[0004]因此，如何改进神经网络的训练方法，使得即使面对稀疏标注，也能够训练神经网络，满足一定的精度要求，对于促进神经网络模型的进一步普及，具有重要的意义。

技术实现思路

[0005]本申请至少提供一种图像处理模型训练方法及相关装置、设备和存储介质。
[0006]本申请第一方面提供了一种图像处理模型训练方法，方法包括：获取样本图像集；其中，样本图像集包括第一样本图像和样本相关图像，第一样本图像标注有目标对象的样本轮廓，样本相关图像的像素点的像素值表示包括第一样本图像中的第一像素点与样本轮廓的样本相关度；利用第一处理模型对第一样本图像进行预测，得到预测图像集；其中，预测图像集包括预测相关图像，预测相关图像包括第一像素点经预测与样本轮廓的预测相关度；基于样本相关图像与预测相关图像，获取第一差异；基于第一差异，调整第一处理模型的网络参数。
[0007]因此，通过利用第一处理模型对第一样本图像进行预测并设置来得到预测相关图像，并基于样本相关图像与预测相关图像的差异来调整第一处理模型的网络参数，实现了对模型的训练。另外，因为样本相关图像是基于第一样本图像而得到的，且第一样本图像的标注信息(样本轮廓)可以认为是弱标签信息，因此也实现了利用弱标签数据对第一处理模型的训练，从而在标注稀少的情况下，通过深度挖掘、学习像素点与轮廓之间的相关度，来尽可能地提升模型精度。
[0008]其中，上述的样本相关度通过结合第一像素点和若干参考维度确定；其中，若干参考维度包括像素差异、像素距离中至少一者。
[0009]因此，对于第一像素点，通过综合包括第一像素点，以及像素差异、像素距离中至少一者，可以获得第一像素点样本轮廓的样本相关度，进而后续可以将样本相关图像作为辅助训练的标注信息，用于后续的训练，提高训练效果。
[0010]其中，上述的第一样本图像为医学图像，目标对象为目标组织，所述目标对象的样本轮廓为所述目标组织的样本轮廓。
[0011]因此，通过限定第一样本图像为医学图像，目标对象为目标组织，以此实现了利用包括目标组织的医学图像来对模型进行训练。
[0012]其中，上述的样本相关图像的获取步骤，包括：基于第一像素点分别与样本轮廓上各个像素点之间的距离，选择一个像素点作为第一像素点的参考像素点；基于第一像素点与参考像素点之间的像素距离和/或像素差异，得到第一像素点与样本轮廓的样本相关度；基于各个第一像素点的样本相关度，得到样本相关图像。
[0013]因此，通过从样本轮廓上各个像素点中选择一个作为参考像素点，使得后续能够基于第一像素点与参考像素点之间的像素距离和/或像素差异来得到第一像素点与样本轮廓的样本相关度，进而得到样本相关图像。
[0014]其中，上述的基于第一像素点分别与样本轮廓上各个像素点之间的距离，选择一个像素点作为第一像素点的参考像素点，包括：在样本轮廓上，选择与第一像素点之间距离最近的像素点，作为第一像素点的参考像素点。
[0015]因此，通过选择与第一像素点之间的距离最近的像素点，作为第一像素点的参考像素点，后续便可以基于二者的像素差异和/或距离差异，确定该第一像素点与样本轮廓的样本相关度。
[0016]其中，上述的第一样本图像分别标注有至少一个目标对象的样本轮廓，第一像素点在各个目标对象的样本轮廓上均选择有参考像素点；上述的基于第一像素点与参考像素点之间的像素距离和/或像素差异，得到第一像素点与样本轮廓的样本相关度，包括：基于第一像素点分别与各个参考像素点之间的像素距离和/或像素差异，得到第一像素点分别与各个目标对象的样本轮廓的子相关度；融合第一像素点的各个子相关度，得到第一像素点的样本相关度。
[0017]因此，通过基于第一像素点分别与各个参考像素点之间的像素距离和/或像素差异，可以得到第一像素点分别与各个目标对象的样本轮廓的子相关度，以此便可基于各个第一像素点的样本相关度，得到样本相关图像，以此实现了多个目标对象的情况下获取样本相关图像。
[0018]其中，上述的第一样本图像分别标注有至少一个目标对象的样本轮廓，第一像素点在各个目标对象的样本轮廓上均选择有参考像素点，且样本图像集包括各个目标对象分别对应的样本相关图像；上述的基于第一像素点与参考像素点之间的像素距离和/或像素差异，得到第一像素点与样本轮廓的样本相关度，包括：基于第一像素点分别与各个参考像素点之间的像素距离和/或像素差异，得到第一像素点分别与各个目标对象的样本轮廓的子相关度；上述的基于各个第一像素点的样本相关度，得到样本相关图像，包括：对于各个目标对象，基于各个第一像素点分别与目标对象的样本轮廓的子相关度，得到目标对象对应的样本相关图像。
[0019]因此，通过分别获取第一像素点分别与各个目标对象的样本轮廓的子相关度，以此可以获得与每个目标对象对应的样本相关图像。
[0020]其中，上述的预测图像集包括各个目标对象分别对应的预测相关图像，上述的基于样本相关图像与预测相关图像，获取第一差异，包括：对于各个目标对象，基于目标对象对应的样本相关图像和预测相关图像，得到目标对象对应的子差异；融合各个目标对象分别对应的子差异，得到第一差异。
[0021]因此，通过基于目标对象对应的样本相关图像和预测相关图像，以此得到每一个目标对象对应的子差异，并且通过各个目标对象分别对应的子差异来得到第一差异，可以使得第一差异能够更加好地反映出样本相关图像和预测相关图像的差异，有助于提高模型的训练效果。
[0022]其中，上述的样本图像集包括样本掩膜图像，样本掩膜图像是基于第一样本图像中标注的目标对象的样本轮廓得到的，且预测图像集还包括第一预测掩膜图像，在上述的基于第一差异，调整第一处理模型的网络参数之前，方法还包括：基于样本掩膜图像与第一预测掩膜图像，获取第二差异；上述的基于第一差异，调整第一处理模型的网络参数，包括：基于第一差异和第二差异，调整第一处理模型的网络参数。
[0023]因此，通过获取第二差异，并基于第一差异和第二差异来调整第一处理模型的网络参数，可以从结合不同的标注信息来对第一处理模型进行训练，有助于提高训练效果。
[0024]其中，上述的基于样本掩膜图像与第一预测掩膜图像，获取第二差异，包括：将第一样本图像中标注的目标对象的样本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种图像处理模型训练方法，其特征在于，包括：获取样本图像集；其中，所述样本图像集包括第一样本图像和样本相关图像，所述第一样本图像标注有目标对象的样本轮廓，所述样本相关图像的像素点的像素值表示所述第一样本图像中的第一像素点与所述样本轮廓的样本相关度；利用第一处理模型对所述第一样本图像进行预测，得到预测图像集；其中，所述预测图像集包括预测相关图像，所述预测相关图像包括所述第一像素点经预测与所述样本轮廓的预测相关度；基于所述样本相关图像与所述预测相关图像，获取第一差异；基于所述第一差异，调整所述第一处理模型的网络参数。2.根据权利要求1所述的方法，其特征在于，所述样本相关度通过结合所述第一像素点和若干参考维度确定；其中，所述若干参考维度包括像素差异、像素距离中至少一者；和/或，所述第一样本图像为医学图像，所述目标对象为目标组织，所述目标对象的样本轮廓为所述目标组织的样本轮廓。3.根据权利要求1或2所述的方法，其特征在于，所述样本相关图像的获取步骤，包括：基于所述第一像素点分别与所述样本轮廓上各个像素点之间的距离，选择一个所述像素点作为所述第一像素点的参考像素点；基于所述第一像素点与所述参考像素点之间的像素距离和/或像素差异，得到所述第一像素点与所述样本轮廓的样本相关度；基于各个所述第一像素点的样本相关度，得到所述样本相关图像。4.根据权利要求3所述的方法，其特征在于，所述基于所述第一像素点分别与所述样本轮廓上各个像素点之间的距离，选择一个所述像素点作为所述第一像素点的参考像素点，包括：在所述样本轮廓上，选择与所述第一像素点之间所述距离最近的像素点，作为所述第一像素点的参考像素点。5.根据权利要求3所述的方法，其特征在于，所述第一样本图像分别标注有至少一个所述目标对象的样本轮廓，所述第一像素点在各个所述目标对象的样本轮廓上均选择有所述参考像素点；所述基于所述第一像素点与所述参考像素点之间的像素距离和/或像素差异，得到所述第一像素点与所述样本轮廓的样本相关度，包括：基于所述第一像素点分别与各个所述参考像素点之间的像素距离和/或像素差异，得到所述第一像素点分别与各个所述目标对象的样本轮廓的子相关度；融合所述第一像素点的各个所述子相关度，得到所述第一像素点的样本相关度。6.根据权利要求3所述的方法，其特征在于，所述第一样本图像分别标注有至少一个所述目标对象的样本轮廓，所述第一像素点在各个所述目标对象的样本轮廓上均选择有所述参考像素点，且所述样本图像集包括各个所述目标对象分别对应的样本相关图像；所述基于所述第一像素点与所述参考像素点之间的像素距离和/或像素差异，得到所述第一像素点与所述样本轮廓的样本相关度，包括：基于所述第一像素点分别与各个所述参考像素点之间的像素距离和/或像素差异，得到所述第一像素点分别与各个所述目标对象的样本轮廓的子相关度；所述基于各个所述第一像素点的样本相关度，得到所述样本相关图像，包括：
对于各个所述目标对象，基于各个所述第一像素点分别与所述目标对象的样本轮廓的子相关度，得到所述目标对象对应的样本相关图像。7.根据权利要求6所述的方法，其特征在于，所述预测图像集包括各个所述目标对象分别对应的预测相关图像，所述基于所述样本相关图像与所述预测相关图像，获取第一差异，包括：对于各个所述目标对象，基于所述目标对象对应的所述样本相关图像和所述预测相关图像，得到所述目标对象对应的子差异；融合各个所述目标对象分别对应的子差异，得到所述第一差异。8.根据权利要求1
‑
7任一项所述的方法，其特征在于，所述样本图像集包括样本...

【专利技术属性】
技术研发人员：胡敏浩，宋涛，冯世祥，张少霆，
申请(专利权)人：上海商汤智能科技有限公司，
类型：发明
国别省市：