一种基于曲线拟合的地铁轨道半自动标注方法技术

本发明专利技术公开了一种基于曲线拟合的地铁轨道半自动标注方法，包括如下步骤：S1，输入待标注的轨道图像；S2，计算拟合曲线方程；S3，根据拟合曲线方程，计算左、右轨道曲线方程；S4，根据左、右轨道曲线方程对轨道图像进行标注；S5，对标注结果进行二值化处理，并输出二值化标注结果。本发明专利技术采用上述技术方案，用户只需鼠标左键点击轨道远端中心点即可自动绘制出轨迹线和输出mask图，相比于传统的人工手动标注方法，很大程度减轻了工作量。相比于边缘检测法提取轨道线，具有速度快、准确率高的优点，并且可以批量操作，提高了轨道标注的效率。

【技术实现步骤摘要】
一种基于曲线拟合的地铁轨道半自动标注方法
本专利技术涉及轨道交通技术，具体涉及轨道交通图像处理技术。
技术介绍
轨道图像的标注是基于深度学习的轨道识别技术中最重要且最基础的工作。由于深度学习需要大量的数据集来训练网络，数据集的制作就是将轨道图像中的轨道线标注出来。面对海量的数据集，人工手动的标注方法任务繁重且效率低，在实际的工程应用中并不可取。基于边缘检测是提取轨道线的一种方法，在环境简单、边界明显的公路车道线或者地面轨道的检测中取得一定的效果，但面对环境复杂的地铁轨道，由于地下光照不均匀，使得轨道图像存在暗影且边界模糊，这些因素导致边缘检测方法在提取地铁轨道线时效果差。
技术实现思路
为了克服现有技术中所存在的上述不足，本专利技术所要解决的技术问题就是提供一种基于曲线拟合的地铁轨道半自动标注方法，可以提高对地铁轨道图像标注的准确率，同时减轻了工作量。为解决上述技术问题，本专利技术采用如下技术方案：一种基于曲线拟合的地铁轨道半自动标注方法，包括如下步骤：S1，输入待标注的轨道图像；S2，计算拟合曲线方程；S3，根据拟合曲线方程，计算左、右轨道曲线方程；S4，根据左、右轨道曲线方程对轨道图像进行标注；S5，对标注结果进行二值化处理，并输出二值化标注结果。优选的，所述步骤S1中，依次自动读取保存在指定文件夹下的所有待标注图片，实现批量化操作。优选的，所述步骤S2中，首先，建立坐标系，坐标原点对应图像中轨道底部中心点Pb0，Pb1、Pb2分别为左、右轨道线底部参考点，Pt1、Pt2分别为轨道透视图像里的视觉消失点，Pt0为消失点Pt1、Pt2的中心点；其次，图中所有参考点的坐标由6个参数来表示，具体如下：Pb0＝(b_m_x,h0)；Pt0＝(t_m_x,t_m_y)；且拟合曲线由6个参数来共同调节，参数具体如下：e：调节拟合曲线远端弯曲程度的常数；b_m_x：调节轨道线底部中心点横坐标；b_dur：调节轨道底部参考点间距；t_m_x：调节消失点中心横坐标；t_m_y：调节消失点中心纵坐标；t_dur：调节轨道远端消失点间距。优选的，6个参数的初始值根据图像预先给出，当固定在车头的相机的位置不发生变化时，拍摄的一系列图片中轨道底部中心点坐标保持不变，此时6个参数的值适用于所有的轨道图像，当轨道底部中心点坐标发生变化时，手动调节相应参数以适应轨道底部中心点坐标发生变化的图片。优选的，通过设置参数滑块的方式，利用滑块手动调节相应参数。优选的，轨道中心线拟合曲线的原始方程如方程式(1)所示：y＝a·xe(1)式中，a为高次拟合曲线系数，e为常数，将中心点坐标(x0,y0)代入方程式(1)，高次拟合曲线系数a可由方程式(2)求得：轨道中心线拟合曲线的最终方程如方程式(3)所示：优选的，所述的步骤S3中，首先，计算轨道横向像素距离，由方程式(4)计算，d＝(h0-y)·w+b(4)式中，h0为输入图像底部纵坐标，w、b为系数，将轨道远端两个消失点以及轨道底部两个参考点的坐标代入方程式(4)即可求得系数w和b；其次，计算左、右轨道线上点的坐标，遍历轨道中心线拟合曲线上各个点的纵坐标，利用方程式(4)可求得拟合曲线上各个点对应的轨道横向像素距离，设拟合曲线上某点的坐标为(x1,y1)，则该点对应的左、右轨道线上点的横坐标可分别表示为方程式(5)与方程式(6)：汇总所有的左、右轨道线上点的坐标即可求得左、右轨道线曲线方程。本专利技术采用上述技术方案，具有如下有益效果：用户只需鼠标左键点击轨道远端中心点，即可获得中心点的坐标，系统按照程序进行处理，即可自动绘制出轨迹线和输出mask图，相比于传统的人工手动标注方法，很大程度减轻了工作量。相比于边缘检测法提取轨道线，具有速度快、准确率高的优点，并且可以批量操作，提高了轨道标注的效率。附图说明图1为本专利技术的流程图。图2为输入待标注的原始图像。图3为建立坐标系示意图。图4为参数滑块界面示意图。图5为轨道中心线拟合曲线示意图。图6为轨道横向像素距离示意图。图7为最终输出的轨道mask图。具体实施方式参考图1所示，一种基于曲线拟合的地铁轨道半自动标注方法，包括以下步骤：1)输入待标注的轨道图像，图2为一张输入的原始图像。本专利技术支持依次自动读取保存在指定文件夹下的所有待标注图片，实现批量化操作。2)计算拟合曲线方程，即对应的轨道中心线曲线。参考图3所示，首先建立坐标系，坐标原点对应图像中轨道底部中心点Pb0，Pb1、Pb2分别为左、右轨道线底部参考点，Pt1、Pt2分别为轨道透视图像里的视觉消失点，Pt0为消失点Pt1、Pt2的中心点。本专利技术中拟合曲线由6个参数来共同调节，参数具体如下：e：调节拟合曲线远端弯曲程度的常数；b_m_x：调节轨道线底部中心点横坐标；b_dur：调节轨道底部参考点间距；t_m_x：调节消失点中心横坐标；t_m_y：调节消失点中心纵坐标；t_dur：调节轨道远端消失点间距。参考图3所示，图中所有参考点的坐标可由6个参数来表示，具体如下：Pb0＝(b_m_x,h0)；Pt0＝(t_m_x,t_m_y)；6个参数的初始值根据图像预先给出，当固定在车头的相机的位置不发生变化时，拍摄的一系列图片中轨道底部中心点坐标保持不变，此时6个参数的值适用于所有的轨道图像。当轨道底部中心点坐标发生变化时，手动调节相应参数以适应轨道底部中心点坐标发生变化的图片。在录制轨道视频时只要保证相机的位置固定，那么同一系列图片中轨道底部中心点坐标是不会发生变化的，之所以要设置这些参数，主要是为了考虑不同系列的图片，比如这次去现场录的是5号线的数据，下一次就可能是2号线的数据，那么不同系列的图片它的轨道底部中心点坐标是不一样的，设置参数是为了适应不同系列的图片，至于将参数调节到何种程度才能适应轨道底部中心点坐标发生变化，就是使得在界面上Pb0和Pb1落在左右轨道上即可。本专利技术通过设置参数滑块，参考图4所示，用户可以利用滑块手动调节相应参数以适应轨道底部中心点坐标发生变化的图片。当然，本领域技术人员可以理解是的，也可以采用直接输入参数的方式。手动调节参数可以理解为一种优化的过程，可以使得最终绘制出的曲线更加贴近真实的轨道曲线，一般来说对于同一系列的图片，不需要调参，只需要用同一组参数就行。轨道中心线拟合曲线的原始方程如(1)所示：y＝a·xe(1)式中，a为高次拟合曲线系数，e为常数，e取值越大曲线远端弯曲程度越大，e取值会给一个初始值，一般先取10去本文档来自技高网...

【技术保护点】
1.一种基于曲线拟合的地铁轨道半自动标注方法，其特征在于包括如下步骤：/nS1，输入待标注的轨道图像；/nS2，计算拟合曲线方程；/nS3，根据拟合曲线方程，计算左、右轨道曲线方程；/nS4，根据左、右轨道曲线方程对轨道图像进行标注；/nS5，对标注结果进行二值化处理，并输出二值化标注结果。/n

【技术特征摘要】
1.一种基于曲线拟合的地铁轨道半自动标注方法，其特征在于包括如下步骤：
S1，输入待标注的轨道图像；
S2，计算拟合曲线方程；
S3，根据拟合曲线方程，计算左、右轨道曲线方程；
S4，根据左、右轨道曲线方程对轨道图像进行标注；
S5，对标注结果进行二值化处理，并输出二值化标注结果。


2.根据权利要求1所述的一种基于曲线拟合的地铁轨道半自动标注方法，其特征在于：所述步骤S1中，依次自动读取保存在指定文件夹下的所有待标注图片，实现批量化操作。


3.根据权利要求1所述的一种基于曲线拟合的地铁轨道半自动标注方法，其特征在于：所述步骤S2中，首先，建立坐标系，坐标原点对应图像中轨道底部中心点Pb0，Pb1、Pb2分别为左、右轨道线底部参考点，Pt1、Pt2分别为轨道透视图像里的视觉消失点，Pt0为消失点Pt1、Pt2的中心点；其次，图中所有参考点的坐标由6个参数来表示，具体如下：
Pb0＝(b_m_x,h0)；






Pt0＝(t_m_x,t_m_y)；






且拟合曲线由6个参数来共同调节，参数具体如下：
e：调节拟合曲线远端弯曲程度的常数；
b_m_x：调节轨道线底部中心点横坐标；
b_dur：调节轨道底部参考点间距；
t_m_x：调节消失点中心横坐标；
t_m_y：调节消失点中心纵坐标；
t_dur：调节轨道远端消失点间距。


4.根据权利要求3所述的一种基于曲线拟合的地铁轨道半自动标注方法，其特征在于：6个参数的初始值根据图像预先给出，当固...

【专利技术属性】
技术研发人员：李康，尤新，程艳丽，
申请(专利权)人：浙江众合科技股份有限公司，
类型：发明
国别省市：浙江;33