[发明专利]一种铁路工机具清点方法及装置

说明书

本发明提供一种铁路工机具清点方法及装置，方法包括：从公共数据库中读取多个工机具样本图像，并对多个所述工机具样本图像进行预处理，通过经预处理后的多个所述工机具样本图像得到训练图像数据集；构建目标检测网络，通过所述训练图像数据集对所述目标检测网络进行训练处理，得到目标检测模型；通过拍摄设备对出工现场进行图像采集，得到待处理图像；根据所述目标检测模型对所述待处理图像进行比对分析，得到遗失工机具类别的名称和数量。本发明实现了对工机具的准确检测与清点，可满足实际应用需求，且不需要人工提取工机具的图像特征，节省了时间的同时也提高了实时性和准确率，为智能铁路运维奠定了技术基础。

技术领域

本发明主要涉及工机具检测领域，具体涉及一种铁路工机具清点方法及装置。

背景技术

铁路是交通系统的重要组成部分。为了保障铁路的安全运行，铁路部门通常在夜间开展运维工作。运维工作中工机具的大量遗失给铁路部门造成了巨大的经济损失。当前的解决方案是对工人领取和归还的工机具进行拍照留存，通过人工对比分析，发现遗失的工机具。然而，人工清点存在主观误差和时效性不足等问题，无法满足人力资源受限情况下铁路运维的智能化要求。基于图像处理技术可实现铁路工机具的自动检测和清点，然而铁路运维通常在夜间进行，而且野外作业环境复杂，拍摄的工机具图像存在背景复杂，光照度低，光照不均等问题。此外，由于铁路工机具种类繁多，工机具图像数据集存在样本稀疏和欠均衡问题。上述问题导致基于图像的铁路工机具自动清点存在精确度低，无法满足实际应用的困难局面。

发明内容

本发明所要解决的技术问题是针对现有技术的不足，提供一种铁路工机具清点方法及装置。

本发明解决上述技术问题的技术方案如下：一种铁路工机具清点方法，包括如下步骤：

从公共数据库中读取多个工机具样本图像，并对多个所述工机具样本图像进行预处理，通过经预处理后的多个所述工机具样本图像得到训练图像数据集；

构建目标检测网络，通过所述训练图像数据集对所述目标检测网络进行训练处理，得到目标检测模型；

通过拍摄设备对出工现场进行图像采集，得到待处理图像；

根据所述目标检测模型对所述待处理图像进行比对分析，得到遗失工机具类别的名称和数量。

本发明解决上述技术问题的另一技术方案如下：一种铁路工机具清点装置，包括：

训练图像数据集处理模块，用于从公共数据库中读取多个工机具样本图像，并对多个所述工机具样本图像进行预处理，通过经预处理后的多个所述工机具样本图像得到训练图像数据集；

目标检测网络处理模块，用于构建目标检测网络，通过所述训练图像数据集对所述目标检测网络进行训练处理，得到目标检测模型；

待处理图像采集模块，用于通过拍摄设备对出工现场进行图像采集，得到待处理图像；

遗失工机具检测模块，用于根据所述目标检测模型对所述待处理图像进行比对分析，得到遗失工机具类别的名称和数量。

本发明的有益效果是：通过对所述多个工机具样本图像进行预处理，通过经处理后的多个工机具样本图像得到训练图像数据集；并构建目标检测网络，通过所述训练图像数据集对所述目标检测网络的训练处理得到目标检测模型，并根据所述目标检测模型对所述待处理图像的比对分析得到遗失工机具类别的名称和数量，实现了对工机具的准确检测与清点，可满足实际应用需求，且不需要人工提取工机具的图像特征，完全由深度神经网络模型端到端、自动地提取特征，能同时完成目标检测与定位两项任务，节省了时间的同时也提高了实时性和准确率，为智能铁路运维奠定了技术基础。

附图说明

图1为本发明一实施例提供的铁路工机具清点方法的流程示意图；