[发明专利]一种拟合弯曲目标轮廓的、装置、计算机设备及存储介质

说明书

本发明公开了拟合弯曲目标轮廓的方法、装置、计算机设备及存储介质。该方法将图片输入分割网络进行特征提取，再进行弯曲目标区域预测，得到连通域图；查找连通域图中的边界点并根据边界点得到边界多边形；对边界多边形进行平滑处理得到弯曲目标的轮廓点线图；根据轮廓点线图中的轮廓点的曲率，筛选出轮廓点线图的左端点集合和右端点集合，并利用左端点集合和右端点集合将轮廓点线图中的轮廓点分类为上边界点和下边界点；从上边界点中筛选出多个控制点，以及从下边界点中筛选出多个控制点，并根据对应的多个控制点分别拟合出弯曲目标的上轮廓线和下轮廓线。本发明提供的方法可以准确地拟合出图片中弯曲目标轮廓，为后续矫正奠定了基础。

技术领域

本发明涉及图像处理技术领域，尤其涉及一种拟合弯曲目标轮廓的、装置、计算机设备及存储介质。

背景技术

图像分割是计算机视觉任务中一项重要的技术。分割网络在分割的过程中，得到的是目标的最大连通域，对于比较规则的目标，通过一些相关算法即可对其轮廓进行拟合；而对于图片中弯曲的目标(如弯曲的文档、弯曲的车道线等)，该连通域无法准确地表达出弯曲目标的轮廓。

发明内容

本发明的目的是提供一种拟合弯曲目标轮廓的方法、装置、计算机设备及存储介质，旨在解决现有分割网络得到的目标的连通域图无法准确地表达出弯曲目标的轮廓的问题。

为解决上述技术问题，本发明的目的是通过以下技术方案实现的：提供一种拟合弯曲目标轮廓的方法，包括：

将图片输入分割网络进行特征提取，再进行弯曲目标区域预测，得到连通域图；

查找所述连通域图中的边界点并根据边界点得到边界多边形；

对所述边界多边形进行平滑处理得到弯曲目标的轮廓点线图；

根据所述轮廓点线图中的轮廓点的曲率，筛选出所述轮廓点线图的左端点集合和右端点集合，并利用所述左端点集合和右端点集合将所述轮廓点线图中的轮廓点分类为上边界点和下边界点；

从所述上边界点中筛选出多个控制点，以及从所述下边界点中筛选出多个控制点，并根据对应的多个控制点分别拟合出弯曲目标的上轮廓线和下轮廓线。

另外，本发明要解决的技术问题是还在于提供一种拟合弯曲目标轮廓的装置，其特征在于，包括：

预处理单元，用于将图片输入分割网络进行特征提取，再进行弯曲目标区域预测，得到连通域图；

查找单元，用于查找所述连通域图中的边界点并根据边界点得到边界多边形；

平滑处理单元，用于对所述边界多边形进行平滑处理得到弯曲目标的轮廓点线图；

分类单元，用于根据所述轮廓点线图中的轮廓点的曲率，筛选出所述轮廓点线图的左端点集合和右端点集合，并利用所述左端点集合和右端点集合将所述轮廓点线图中的轮廓点分类为上边界点和下边界点；

拟合单元，用于从所述上边界点中筛选出多个控制点，以及从所述下边界点中筛选出多个控制点，并根据对应的多个控制点分别拟合出弯曲目标的上轮廓线和下轮廓线。

另外，本发明实施例又提供了一种计算机设备，其包括存储器、处理器及存储在所述存储器上并可在所述处理器上运行的计算机程序，所述处理器执行所述计算机程序时实现上述第一方面所述的拟合弯曲目标轮廓的方法。

另外，本发明实施例还提供了一种计算机可读存储介质，其中所述计算机可读存储介质存储有计算机程序，所述计算机程序当被处理器执行时使所述处理器执行上述第一方面所述的拟合弯曲目标轮廓的方法。