[发明专利]边缘提取方法、装置、电子设备及存储介质

说明书

本公开实施例公开了一种边缘提取方法、装置、电子设备及存储介质，该方法包括：获取目标待提取图像；将目标待提取图像输入至目标边缘提取模型中，得到与目标待提取图像对应的目标边缘掩膜图像；目标边缘提取模型基于下述方法训练得到：获取样本初始待提取图像以及样本初始边缘掩膜图像；对样本初始待提取图像进行图像增强处理，并对样本初始边缘掩膜图像进行图像增强处理根据图像增强处理后的样本初始待提取图像和样本初始边缘掩膜图像对初始深度学习模型进行训练，得到目标边缘提取模型。本公开实施例的技术方案，通过图像增强处理后的样本初始待提取图像以及样本初始边缘掩膜图像训练得到目标边缘提取模型，能够更精准地提取图像中的边缘信息。

技术领域

本公开实施例涉及图像处理技术领域，尤其涉及一种边缘提取方法、装置、电子设备及存储介质。

背景技术

图像边缘作为图像基本的特征，集中了大量的图像信息。图像边缘检测是图像处理和计算机视觉中的基本问题。图像边缘通常存在于目标、背景和区域之间，因此，对于图像边缘的检测和提取具有较大的难度。

目前，基于现有的图像边缘检测和提取技术对图像边缘进行检测和提取时，存在着图像边缘提取结果粗糙，不够精细的问题。

发明内容

本公开实施例提供了一种边缘提取方法、装置、电子设备及存储介质，以实现更为精准地提取图像中的边缘信息的效果。

第一方面，本公开实施例提供了一种边缘提取方法，该方法包括：

获取目标待提取图像；

将所述目标待提取图像输入至目标边缘提取模型中，得到与所述目标待提取图像对应的目标边缘掩膜图像；

其中，所述目标边缘提取模型基于下述方法训练得到：

获取样本初始待提取图像以及与所述样本初始待提取图像对应的样本初始边缘掩膜图像；

对所述样本初始待提取图像进行图像增强处理，得到目标尺寸的样本目标待提取图像，并对所述样本初始边缘掩膜图像进行图像增强处理，得到所述目标尺寸的样本目标边缘掩膜图像；

根据所述样本目标待提取图像以及与所述样本目标待提取图像对应的样本目标边缘掩膜图像对初始深度学习模型进行训练，得到目标边缘提取模型。

第二方面，本公开实施例还提供了一种边缘提取装置，该装置包括：

图像获取模块，用于获取目标待提取图像；

边缘提取模块，用于将所述目标待提取图像输入至目标边缘提取模型中，得到与所述目标待提取图像对应的目标边缘掩膜图像；

其中，所述目标边缘提取模型基于模型训练装置得到，所述模型训练装置包括：

样本获取模块，用于获取样本初始待提取图像以及与所述样本初始待提取图像对应的样本初始边缘掩膜图像；

样本增强模块，用于对所述样本初始待提取图像进行图像增强处理，得到目标尺寸的样本目标待提取图像，并对所述样本初始边缘掩膜图像进行图像增强处理，得到所述目标尺寸的样本目标边缘掩膜图像；

模型训练模块，用于根据所述样本目标待提取图像以及与所述样本目标待提取图像对应的样本目标边缘掩膜图像对初始深度学习模型进行训练，得到目标边缘提取模型。

第三方面，本公开实施例还提供了一种电子设备，该电子设备包括：

一个或多个处理器；

存储装置，用于存储一个或多个程序，

当所述一个或多个程序被所述一个或多个处理器执行，使得所述一个或多个处理器实现本公开任意实施例所提供的边缘提取方法。