[发明专利]图像处理方法、装置、电子设备及存储介质

说明书

本公开提供了一种图像处理方法、装置、电子设备及存储介质，涉及人工智能领域，进一步涉及计算机视觉领域和深度学习领域，以至少解决相关技术中难以提取到完整道路的技术问题的技术问题。具体实现方案为：获取目标道路的骨架图像，其中，骨架图像至少包括：目标道路的至少一条道路骨架；对骨架图像进行矢量处理，生成与至少一条道路骨架对应的目标点集合；基于目标点集合对骨架图像进行修复，生成目标图像。

技术领域

本公开涉及人工智能领域，进一步涉及计算机视觉领域和深度学习领域，尤其涉及一种图像处理方法、装置、电子设备及存储介质。

背景技术

目前，一般是通过预测模型对道路图像进行预测，提取道路图像中的道路，但是，由于道路的特殊性，即一般会有树木等遮挡物对道路进行遮挡，导致提取出的道路会出现断开的情况，从而难以得到完整的道路。

发明内容

本公开提供了一种图像处理方法、装置、电子设备以及存储介质，以至少解决相关技术中难以从提取到完整道路的技术问题。

根据本公开的一方面，提供了一种图像处理方法，包括：获取目标道路的骨架图像，其中，骨架图像至少包括：目标道路的至少一条道路骨架；对骨架图像进行矢量处理，生成与至少一条道路骨架对应的目标点集合；基于目标点集合对骨架图像进行修复，生成目标图像。

可选地，基于目标点集合对骨架图像进行修复，生成目标图像，包括：获取目标点集合中的至少一对目标点，其中，每对目标点中的两个点之间的距离小于预设距离；连接骨架图像中的每对目标点，生成目标图像。

可选地，对骨架图像进行矢量处理，生成与至少一条道路骨架对应的目标点集合，包括：利用第一预设函数对骨架图像进行矢量处理，生成目标无向图；从目标无向图中获取至少一条道路骨架的第一点集合，其中，第一点集合为每条道路骨架中端点的集合；从目标无向图中获取至少一条道路骨架的第二点集合，其中，第二点集合为每条道路骨架与至少一条道路骨架中的其他道路骨架中交叉点的集合；基于第一点集合和第二点集合，生成目标点集合。

可选地，在获取目标道路的骨架图像之前，该方法还包括：获取目标道路的原始图像；利用语义分割模型对原始图像进行预测，基于预测结果获取目标道路的第一掩膜图像；利用第二预设函数从第一掩膜图像中提取目标道路的骨架图像。

可选地，基于目标点集合对骨架图像进行修复，生成目标图像，包括：基于目标点集合对骨架图像进行修复，生成修复后的骨架图像；对修复后的骨架图像进行膨胀操作，生成第二掩膜图像；将第二掩膜图像标注在原始图像中，生成目标图像。

可选地，在对修复后的骨架图像进行膨胀操作，生成第二掩膜图像之后，该方法还包括：获取对目标道路对应的骨架图像的修复次数；响应于修复次数小于或等于预设次数的情况下，将第二掩膜图像作为第一掩膜图像，执行利用第二预设函数从第一掩膜图像中提取目标道路的骨架图像的步骤。

根据本公开的又一方面，提供了一种图像处理装置，包括：获取模块，用于获取目标道路的骨架图像，其中，骨架图像至少包括：目标道路的至少一条道路骨架；处理模块，用于对骨架图像进行矢量处理，生成与至少一条道路骨架对应的目标点集合；修复模块，用于基于目标点集合对骨架图像进行修复，生成目标图像。

根据本公开的又一方面，提供了一种电子设备，包括：至少一个处理器；以及与至少一个处理器通信连接的存储器；其中，存储器存储有可被至少一个处理器执行的指令，指令被至少一个处理器执行，以使至少一个处理器能够执行本公开提出的图像处理方法。

根据本公开的又一方面，提供了一种存储有计算机指令的非瞬时计算机可读存储介质，其中，计算机指令用于使计算机执行本公开提出的图像处理方法。

根据本公开的又一方面，提供了一种计算机程序产品，包括计算机程序，计算机程序在被处理器执行本公开提出的图像处理方法。