公开(公告)号：CN111899227A

公开(公告)日：2020-11-06

申请号：CN202010640270.4

申请日：2020-07-06

Applicant: 北京交通大学

Inventor： 秦勇 ,  崔京 ,  吴云鹏 ,  刘嘉豪 ,  曹先彬 ,  樊邦奎 ,  李齐贤 ,  陈平

IPC: G06T7/00 ,  G06K9/62 ,  G06N3/04

Abstract: 本发明提供了一种基于无人机作业的铁路扣件缺陷自动采集辨识方法。该方法包括：对无人机拍摄的铁路扣件图像数据进行增强处理，得到训练集、验证集和测试集；利用CYOLO算法构建网络模型，利用训练集中的图像数据对CYOLO网络模型进行训练，利用验证集中的图像数据对训练完成的CYOLO网络模型进行比对检测，得到训练好的CYOLO网络模型；将训练好的CYOLO网络模型作为铁路扣件缺陷图像的检测模型，将测试集中的图像数据输入到铁路扣件缺陷图像的检测模型中，实现图像数据的铁路扣件缺陷目标检测。本发明将CYOLO算法应用到铁路扣件缺陷巡检中，实现扣件缺陷自动识别，提高检测效率，可以实现对铁路无人机图像扣件缺陷的坐标定位和分类。