一种图像分割方法、装置、电子设备及存储介质

本申请公开了一种图像分割方法、装置、电子设备及存储介质，方法包括：获取原始图像；对原始图像进行处理得到训练图像数据集；将训练图像数据集输入原始图像分割预测模型进行训练，得到目标图像分割预测模型；原始图像分割预测模型的目标局部自注意力模块基于边缘检测算子计算的第一梯度强度概率构建，原始图像分割预测模型的目标下采样模块基于边缘检测算子计算的第二梯度强度概率构建；将待预测图像的预测图像块输入目标图像分割预测模型进行预测，输出图像分割预测结果；根据图像分割预测结果对待预测图像进行分割，得到目标分割图像。本申请能够提高图像分割的精确率、降低模型复杂度并降低对计算资源的消耗，可广泛应用于图像处理技术领域。