[发明专利]神经网络训练与图像处理方法及装置

说明书

本公开涉及一种神经网络训练与图像处理方法及装置，该神经网络可以用于对医学影像图像进行标注，比如对心脏、肝脏、肺部以及病理图像进行标注。所述方法包括：根据第一样本图像，对神经网络进行训练，获得第一训练状态的神经网络；根据第二样本图像、第一样本图像，对第一训练状态的神经网络进行训练，获得第二训练状态的神经网络。根据本公开的实施例的神经网络训练方法，可通过具有不同标注的样本图像逐步对神经网络进行训练，以增强神经网络处理具有不同标注的样本图像的能力，提高神经网络的精度和泛化能力。

技术领域

本公开涉及计算机技术领域，尤其涉及一种神经网络训练与图像处理方法及装置。

背景技术

在相关技术中，对样本的标注精度不同，例如，根据标注的精度，可分为无标注的样本、具有类别标注的样本、具有检测框标注的样本和具有分割轮廓标注的样本。标注的信息量不同，则标注成本也不同，例如，分割轮廓标注或检测框标注的信息量大于类别标注，标注成本也大于类别标注。因此，具有分割轮廓标注的样本的数量少于具有类别标注的样本的数量。在神经网络训练过程中，标注信息量大的样本数量较少，使得训练后的神经网络的精度与泛化能力不足。使用标注信息量小的样本进行训练，使得训练后的神经网络的性能不足。

发明内容

本公开提出了一种神经网络训练与图像处理方法及装置。

根据本公开的一方面，提供了一种神经网络训练方法，包括：根据第一样本图像，对神经网络进行训练，获得第一训练状态的神经网络，所述第一样本图像具有区域标注信息；根据第二样本图像、所述第一样本图像，对所述第一训练状态的神经网络进行训练，获得第二训练状态的神经网络，所述第二样本图像具有类别标注信息。

根据本公开的实施例的神经网络训练方法，可通过具有不同标注的样本图像逐步对神经网络进行训练，以增强神经网络处理具有不同标注的样本图像的能力，提高神经网络的精度和泛化能力。

在一种可能的实现方式中，根据第二样本图像、所述第一样本图像，对所述第一训练状态的神经网络进行训练，获得第二训练状态的神经网络，包括：使用第i个训练周期的神经网络对第二样本图像进行检测处理，获得第二样本图像中的第一预测区域，其中，i为正整数，所述第i个训练周期为将所述第一训练状态的神经网络训练成为第二训练状态的神经网络的多个训练周期中的任意一个训练周期；使用具有类别标注信息和所述第一预测区域的第二样本图像以及具有区域标注信息的第一样本图像训练所述第i个训练周期的神经网络，获得第i+1个训练周期的神经网络；在所述神经网络满足第一训练条件时，获得所述第二训练状态的神经网络。

通过这种方式，可通过具有类别标注信息的第二样本图像训练神经网络，补充第一样本图像的数量，提高神经网络的检测精度。

在一种可能的实现方式中，使用第i个训练周期的神经网络对具有类别标注信息的第二样本图像进行检测处理，获得第二样本图像中的第一预测区域，包括：使用所述第i个训练周期的神经网络对具有类别标注信息的第二样本图像进行检测处理，获得第二样本图像中的第一样本区域；在所述第一样本区域中，确定出置信度大于或等于第一置信度阈值的第二样本区域；根据所述类别标注信息，在所述第二样本区域中确定出所述第一预测区域。

在一种可能的实现方式中，所述方法还包括：根据未标注的第三样本图像、所述第二样本图像、所述第一样本图像，对所述第二训练状态的神经网络进行训练，获得训练后的神经网络。

通过这种方式，可通过数量较多的未标注的第三样本图像和具有类别标注信息的第二样本图像训练神经网络，降低了对样本图像的标注成本。