[发明专利]一种轨面语义分割方法及装置

说明书

本申请提供了一种轨面语义分割方法及装置，其中，该方法包括：获取轨道图像；将所述轨道图像输入至预先训练好的轨面语义分割模型中，获取所述轨道图像中每个像素点对应的分类概率，所述分类概率用于表征每个像素点是轨面特征的概率；根据所述轨道图像中每个像素点对应的分类概率和预设分类概率阈值，得到所述轨道图像的轨面分割图像，所述轨面分割图像标记有所述轨道图像中的轨面区域以及背景区域。本申请实施例能够通过图像分割技术从列车行驶前方的图像中分割出轨面部分，提升轨面与其他区域的视觉差别，从而可以帮助驾驶员更好的辨别轨道上是否有障碍物，提高辨别的准确率。

技术领域

本申请涉及轨道交通技术领域，具体而言，涉及一种轨面语义分割方法及装置。

背景技术

目前，在轨道交通中，列车行驶区域常有障碍物，如行人、车辆、或者其他物品等，极易引发交通事故。目前列车在行驶过程中，主要靠驾驶员目视行驶区域，人为判断轨道上是否有障碍物。但是由于轨道的颜色和其他背景或障碍物的颜色相差不大，且列车行驶速度大，而人眼视距较短，因此仅仅靠人眼去辨别是否有障碍物容易出现漏判的情况，导致辨别的准确率较低。

发明内容

有鉴于此，本申请实施例的目的在于提供一种轨面语义分割方法及装置，能够通过图像分割技术从列车行驶前方的图像中分割出轨面部分，提升轨面与其他区域的视觉差别，从而可以帮助驾驶员更好的辨别轨道上是否有障碍物，提高辨别的准确率。

第一方面，本申请实施例提供了一种轨面语义分割方法，包括：

获取轨道图像；

将所述轨道图像输入至预先训练好的轨面语义分割模型中，获取所述轨道图像中每个像素点对应的分类概率，所述分类概率用于表征每个像素点是轨面特征的概率；

根据所述轨道图像中每个像素点对应的分类概率和预设分类概率阈值，得到所述轨道图像的轨面分割图像，所述轨面分割图像标记有所述轨道图像中的轨面区域以及背景区域。

结合第一方面，本申请实施例提供了第一方面的第一种可能的实施方式，其中，通过下述方式获得所述预先训练好的轨面语义分割模型：

获取多个样本轨道图像，以及与所述样本轨道图像对应的样本标注图像，所述样本标注图像标记有所述样本轨道图像中的轨面区域以及背景区域；

将所述样本轨道图像输入至预先构建的轨面语义分割模型中，获取所述样本轨道图像中每个像素点对应的分类概率；

根据所述样本轨道图像中每个像素点对应的分类概率以及与各个样本轨道图像对应的样本标注图像，对所述轨面语义分割模型进行本轮训练；

经过对所述轨面语义分割模型进行多轮训练，得到所述预先训练好的轨面语义分割模型。

结合第一方面的第一种可能的实施方式，本申请实施例提供了第一方面的第二种可能的实施方式，其中，所述根据所述样本轨道图像中每个像素点对应的分类概率以及与各个样本轨道图像对应的样本标注图像，对所述轨面语义分割模型进行本轮训练，包括：

针对每个所述样本轨道图像，根据该样本轨道图像中每个像素点对应的分类概率以及与该样本轨道图像对应的样本标注图像，获得该样本轨道图像中每个像素点对应的交叉熵损失；

根据该样本轨道图像中每个像素点对应的交叉熵损失，调整所述轨面语义分割模型的参数；

直至所有样本轨道图像完成本轮的训练后，确定完成对所述轨面语义分割模型的本轮训练。

结合第一方面，本申请实施例提供了第一方面的第三种可能的实施方式，其中，所述根据所述轨道图像中每个像素点对应的分类概率和预设分类概率阈值，得到所述轨道图像的轨面分割图像，包括：