图像处理模型的训练方法、装置、电子设备及存储介质与流程

本申请涉及人工智能技术，尤其涉及一种图像处理模型的训练方法、装置、电子设备及存储介质。

背景技术：

1、人工智能(artificial intelligence，ai)是利用数字计算机或者数字计算机控制的机器模拟、延伸和扩展人的智能，感知环境、获取知识并使用知识获得最佳结果的理论、方法、技术及应用系统。计算机视觉技术(computer vision，cv)计算机视觉是一门研究如何使机器“看”的科学，更进一步的说，就是指用摄影机和电脑代替人眼对目标进行识别、定位和测量等机器视觉，并进一步做图形处理，使电脑处理成为更适合人眼观察或传送给仪器检测的图像。

2、针对三维图像的分割处理，相关技术中通常将三维图像转换为二维图像进行处理，例如光学相干断层扫描(optical coherence tomography，oct)图像，针对图像中的每个切片图像进行处理，在训练相关的图像处理模型的过程中，通常需要对训练样本进行人工标注，消耗了大量的时间以及计算资源，

3、相关技术，针对提升分割三维图像的准确性暂无较好的解决方法。

技术实现思路

1、本申请实施例提供一种图像处理模型的训练方法、装置、电子设备及计算机可读存储介质、计算机程序产品，能够提升针对三维图像进行分割的准确性。

2、本申请实施例的技术方案是这样实现的：

3、本申请实施例提供一种图像处理模型的训练方法，包括：

4、获取多个三维图像样本以及标注数据，其中，所述三维样本的类型包括标注图像样本和未标注图像样本，所述标注数据包括所述标注图像样本的实际分割位置；

5、基于所述多个三维图像样本调用所述图像处理模型进行图像分割处理，得到第一预测分割位置，并基于所述第一预测分割位置与所述标注数据之间的第一损失，更新所述图像处理模型；

6、将所述未标注图像样本的第一预测分割位置，作为所述未标注图像样本的实际分割位置，其中，所述第一预测分割位置是调用所述图像处理模型生成的；

7、基于每个所述三维图像样本的实际分割位置，对所述第一损失进行更新，得到第二损失；

8、基于所述第二损失更新基于所述第一损失更新后的所述图像处理模型。

9、本申请实施例提供一种图像处理方法，所述方法包括：

10、获取待分割的三维图像；

11、基于所述三维图像调用图像处理模型进行图像分割处理，得到所述三维图像中的分割位置，其中，所述图像处理模型是通过本申请实施例的图像处理模型的训练方法得到的。

12、本申请实施例提供一种图像处理模型的训练装置，所述装置包括：

13、样本获取模块，配置为获取多个三维图像样本以及标注数据，其中，所述三维样本的类型包括标注图像样本和未标注图像样本，所述标注数据包括所述标注图像样本的实际分割位置；

14、预训练模块，配置为基于所述多个三维图像样本调用所述图像处理模型进行图像分割处理，得到第一预测分割位置，并基于所述第一预测分割位置与所述标注数据之间的第一损失，更新所述图像处理模型；

15、模型更新模块，配置为将所述未标注图像样本的第一预测分割位置，作为所述未标注图像样本的实际分割位置，其中，所述第一预测分割位置是调用所述图像处理模型生成的；

16、所述模型更新模块，还配置为基于每个所述三维图像样本的实际分割位置，对所述第一损失进行更新，得到第二损失；

17、所述模型更新模块，还配置为基于所述第二损失更新基于所述第一损失更新后的所述图像处理模型。

18、本申请实施例提供一种图像处理装置，所述图像处理装置包括：

19、图像接收模块，配置为获取待分割的三维图像；

20、图像处理模块，配置为基于所述三维图像调用图像处理模型进行图像分割处理，得到所述三维图像中的分割位置，其中，所述图像处理模型是通过本申请实施例的图像处理模型的训练方法得到的。

21、本申请实施例提供一种电子设备，包括：

22、存储器，用于存储计算机可执行指令；

23、处理器，用于执行所述存储器中存储的计算机可执行指令时，实现本申请实施例提供的图像处理模型的训练方法。

24、本申请实施例提供一种计算机可读存储介质，存储有计算机可执行指令，用于引起处理器执行时，实现本申请实施例提供的图像处理模型的训练方法。

25、本申请实施例提供一种计算机程序产品，包括计算机程序或计算机可执行指令，所述计算机程序或计算机可执行指令被处理器执行时实现本申请实施例提供的图像处理模型的训练方法。

26、本申请实施例具有以下有益效果：

27、通过两阶段地进行训练，提升了训练图像处理模型的准确性，通过稀疏标注的方式基于预测得到的第一预测分割位置对未标注的训练数据进行标注，以更新图像处理模型对应的损失，并通过更新的损失对图像处理模型进行优化训练，提升了训练得到的图像处理模型的准确性，节约了标注训练数据所需的计算资源。

技术特征：

1.一种图像处理模型的训练方法，其特征在于，所述方法包括：

2.根据权利要求1所述的方法，其特征在于，所述第一损失包括：第一对齐损失以及第一分割损失；

3.根据权利要求2所述的方法，其特征在于，所述获取所述三维图像样本的对齐特征，包括：

4.根据权利要求3所述的方法，其特征在于，所述图像处理模型包括：特征提取网络、对齐网络、空间位移模块、分割网络；

5.根据权利要求4所述的方法，其特征在于，所述基于每个所述对齐特征与所述标注数据确定第一对齐损失，包括：

6.根据权利要求5所述的方法，其特征在于，所述获取两个所述三维图像的对齐特征之间的子互相关损失，包括：

7.根据权利要求5所述的方法，其特征在于，所述分割处理的结果包括：所述三维图像中每个位置的分布概率、所述第一预测分割位置；

8.根据权利要求7所述的方法，其特征在于，所述基于所述第一对齐损失与所述第一分割损失对所述图像处理模型进行参数更新处理，包括：

9.根据权利要求8所述的方法，其特征在于，所述分割网络包括第一输出头以及第二输出头；所述第二输出头用于输出所述三维图像中每个位置的分布概率，所述第一输出头用于输出所述第一预测分割位置；

10.根据权利要求1所述的方法，其特征在于，

11.根据权利要求1所述的方法，其特征在于，

12.一种图像处理方法，其特征在于，所述方法包括：

13.根据权利要求12所述的方法，其特征在于，所述基于所述三维图像调用图像处理模型进行图像分割处理，得到所述三维图像中的分割位置，包括：

14.一种图像处理模型的训练装置，其特征在于，所述图像处理模型的训练装置包括：

15.一种图像处理装置，其特征在于，所述图像处理装置包括：

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

17.一种计算机可读存储介质，存储有计算机可执行指令，其特征在于，所述计算机可执行指令被处理器执行时，实现权利要求1至11任一项所述的图像处理模型的训练方法，或权利要求12至13任一项所述的图像处理方法。

18.一种计算机程序产品，包括计算机程序或计算机可执行指令，其特征在于，所述计算机程序或计算机可执行指令被处理器执行时，实现权利要求1至11任一项所述的图像处理模型的训练方法，或权利要求12至13任一项所述的图像处理方法。

技术总结
本申请提供了一种图像处理模型的训练方法、装置、电子设备及存储介质；方法包括：获取多个三维图像样本以及标注数据，三维样本的类型包括标注图像样本和未标注图像样本，标注数据包括标注图像样本的实际分割位置；基于多个三维图像样本调用图像处理模型进行图像分割处理，得到第一预测分割位置，并基于第一预测分割位置与标注数据之间的第一损失，更新图像处理模型；将未标注图像样本的第一预测分割位置，作为未标注图像样本的实际分割位置；基于每个三维图像样本的实际分割位置，对第一损失进行更新，得到第二损失；基于第二损失更新基于第一损失更新后的图像处理模型。通过本申请，能够提升针对三维图像进行分割的准确性。

技术研发人员：魏东,刘洪,卢东焕,黄雅雯,王连生,郑冶枫
受保护的技术使用者：腾讯科技（深圳）有限公司
技术研发日：
技术公布日：2024/1/16