[发明专利]医学图像分割模型的训练方法、介质及电子设备

说明书

本发明提供一种医学图像分割模型的训练方法、介质及电子设备。所述方法包括：对所述医学图像分割模型进行第一阶段的训练，其中，所述第一阶段的训练包括多轮训练；对所述医学图像分割模型进行第二阶段的训练，以完成对所述医学图像分割模型的训练，其中，所述第二阶段的训练包括多轮训练。所述方法可以直接利用所述普通样本和所述困难样本完成对所述医学图像分割模型的训练。

技术领域

本发明涉及一种训练方法，特别是涉及一种医学图像分割模型的训练方法、介质及电子设备。

背景技术

随着人工智能技术的不断发展，利用基于AI的医学图像分割模型对医学图像进行处理进而辅助医生进行诊断已成为相关领域的惯用手段。医学图像分割模型在实际应用之前需要进行训练以便获取对医学图像进行处理的能力，然而，医学图像的尺寸往往都比较大，例如，肝脏CT图像的尺寸通常为512×512。而在医学图像分割模型的训练过程中，特别是在3D医学图像分割模型的训练过程中，受到GPU显存大小的限制，通常很难直接将医学图像按照原始大小直接输入医学图像分割模型来对其进行训练。因此，如何提供一种能够适用于医学图像分割模型的训练方法已成为相关领域技术人员亟需解决的技术问题之一。

发明内容

鉴于以上所述现有技术的缺点，本发明的目的在于提供一种医学图像分割模型的训练方法、介质及电子设备，用于解决现有技术中存在的上述问题。

为实现上述目的及其他相关目的，本发明的第一方面提供一种医学图像分割模型的训练方法，用于对一医学图像分割模型进行训练，所述医学图像分割模型的训练方法包括：获取多个训练图像；对所述医学图像分割模型进行第一阶段的训练，其中，所述第一阶段的训练包括多轮训练；对所述医学图像分割模型进行第二阶段的训练，以完成对所述医学图像分割模型的训练，其中，所述第二阶段的训练包括多轮训练；在所述第一阶段的每轮训练中，对所述医学图像分割模型进行训练的方法包括：根据所述训练图像获取普通样本，其中，所述普通样本的尺寸小于所述训练图像的尺寸；利用所述普通样本对所述医学图像分割模型进行训练；在所述第二阶段的每轮训练中，对所述医学图像分割模型进行训练的方法包括：根据所述训练图像获取普通样本或者从困难样本库中获取困难样本，其中，所述普通样本的尺寸与所述困难样本相同；根据所述普通样本或所述困难样本对所述医学图像分割模型进行训练。

于所述第一方面的一实施例中，根据所述训练图像获取普通样本的实现方法包括：从所述训练图像中随机选取特定尺寸的区域作为所述普通样本。

于所述第一方面的一实施例中，在所述第一阶段的训练完成以后，或者，在所述第二阶段的每轮训练完成以后，所述医学图像分割模型的训练方法还包括：初始化所述困难样本库；利用所述医学图像分割模型对各所述训练图像进行处理，以得到各所述训练图像的预测结果；根据各所述训练图像的预测结果及真实标注结果对所述困难样本库进行更新。

于所述第一方面的一实施例中，根据各所述训练图像的预测结果及真实标注结果对所述困难样本库进行更新的实现方法包括：根据各所述训练图像的预测结果及真实标注结果，获取各所述训练图像对应的预测效果；对于预测效果较差的第一训练图像，获取各所述第一训练图像对应的困难样本并存入所述困难样本库。

于所述第一方面的一实施例中，获取所述第一训练图像对应的困难样本的实现方法包括：根据所述第一训练图像的预测结果及真实标注结果，获取所述第一训练图像的误差图；对所述第一训练图像的误差图进行池化，并根据池化结果选取误差最大的区域作为所述第一训练图像对应的困难样本，在所述池化中所采用的池化核的尺寸与所述普通样本的尺寸相同。

于所述第一方面的一实施例中，根据所述训练图像获取普通样本或者从困难样本库中获取困难样本的实现方法包括：生成一随机数；当所述随机数的数值满足预设条件时，根据所述训练图像获取所述普通样本，否则，从所述困难样本库中获取所述困难样本。

于所述第一方面的一实施例中，所述训练图像的尺寸大于或等于512×512。