[发明专利]图像处理、图像处理模型的训练方法、装置、设备及介质

说明书

本申请公开了一种图像处理、图像处理模型的训练方法、装置、设备及介质，涉及人工智能技术领域中的计算机视觉技术。该图像处理方法包括：获取参考对象的待处理图像和目标图像处理模型，目标图像处理模型基于参考对象的样本图像和样本图像对应的标签训练得到，样本图像对应的标签用于指示样本图像中的第一样本像素点；调用目标图像处理模型对待处理图像的表征图像进行概率预测，得到第一概率图；基于第一概率图，获取目标对象的图像。此种方式，获取目标图像的效率较高，目标图像处理模型能够较为精准地识别出图像中的对应的对象类别为目标对象的像素点，调用目标图像处理模型获取的第一概率图的可靠性较高，目标对象的图像的精度较高。

技术领域

本申请实施例涉及人工智能技术领域，特别涉及一种图像处理、图像处理模型的训练方法、装置、设备及介质。

背景技术

随着人工智能技术的发展，对图像进行处理的应用场景越来越多。其中，一种应用场景为：根据参考对象的图像获取目标对象的图像，以利用目标对象的图像执行配准以及图像标注等任务。其中，目标对象为参考对象对应的满足参考条件的对象，例如，满足参考条件是指轮廓上未附有异物且内部不存在孔洞。

目前，获取目标对象的图像的过程需要人工参与，效率较低，受人工的主观因素影响较大，获取的目标对象的图像的精度较差。

发明内容

本申请实施例提供了一种图像处理、图像处理模型的训练方法、装置、设备及介质，可用于提高获取目标对象的图像的效率以及提高获取的目标对象的图像的精度。所述技术方案如下：

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

获取参考对象的待处理图像和目标图像处理模型，所述目标图像处理模型基于所述参考对象的样本图像和所述样本图像对应的标签训练得到，所述样本图像对应的标签用于指示所述样本图像中的第一样本像素点，所述第一样本像素点对应的对象类别为所述参考对象对应的满足参考条件的目标对象；

调用所述目标图像处理模型对所述待处理图像的表征图像进行概率预测，得到第一概率图，所述第一概率图中的像素点的像素值用于指示所述像素点对应的对象类别为所述目标对象的概率；

基于所述第一概率图，获取所述目标对象的图像。

还提供了一种图像处理模型的训练方法，所述方法包括：

获取参考对象的样本图像和所述样本图像对应的标签，所述样本图像对应的标签用于指示所述样本图像中的第一样本像素点，所述第一样本像素点对应的对象类别为所述参考对象对应的满足参考条件的目标对象；

调用初始图像处理模型对所述样本图像的表征图像进行概率预测，得到样本概率图，所述样本概率图中的像素点的像素值用于指示所述像素点对应的对象类别为所述目标对象的概率；

基于所述样本概率图和所述样本图像对应的标签，对所述初始图像处理模型进行训练，得到目标图像处理模型。

另一方面，提供了一种图像处理装置，所述装置包括：

第一获取单元，用于获取参考对象的待处理图像和目标图像处理模型，所述目标图像处理模型基于所述参考对象的样本图像和所述样本图像对应的标签训练得到，所述样本图像对应的标签用于指示所述样本图像中的第一样本像素点，所述第一样本像素点对应的对象类别为所述参考对象对应的满足参考条件的目标对象；

第一处理单元，用于调用所述目标图像处理模型对所述待处理图像的表征图像进行概率预测，得到第一概率图，所述第一概率图中的像素点的像素值用于指示所述像素点对应的对象类别为所述目标对象的概率；

第二获取单元，用于基于所述第一概率图，获取所述目标对象的图像。

在一种可能实现方式中，所述装置还包括：