发明授权
- 专利标题: 一种轨道缺陷检测系统鲁棒性提升方法
-
申请号: CN202410315833.0申请日: 2024-03-20
-
公开(公告)号: CN117934819B公开(公告)日: 2024-07-05
- 发明人: 李栋 , 崔晓彤 , 范建国 , 郑伟 , 王烨 , 王睿 , 江周娴 , 冯海燕 , 付功云 , 王婷 , 杨喆
- 申请人: 中铁第六勘察设计院集团有限公司 , 北京交通大学
- 申请人地址: 天津市滨海新区自贸试验区(空港经济区)中环西路36号;
- 专利权人: 中铁第六勘察设计院集团有限公司,北京交通大学
- 当前专利权人: 中铁第六勘察设计院集团有限公司,北京交通大学
- 当前专利权人地址: 天津市滨海新区自贸试验区(空港经济区)中环西路36号;
- 代理机构: 天津睿勤专利代理事务所
- 代理商 张伟凤
- 主分类号: G06V10/25
- IPC分类号: G06V10/25 ; G06Q10/20 ; G06Q50/40 ; G06Q50/50 ; G06V10/774 ; G06V10/776 ; G06V10/82
摘要:
本发明提供了一种轨道缺陷检测系统鲁棒性提升方法,首先对采集到的轨道缺陷数据进行预处理,按照8:1:1的比例划分为训练集、验证集和测试集;基于YOLOv5目标检测模型生成轨道缺陷检测模型,采用一系列指标对轨道缺陷检测模型性能进行评估,通过训练集进行初训练得到初训练模型;采用变异生成算法生成更具多样性的训练样本,将训练集上生成的多样性数据与原始训练集合并来对初训练模型进行重训练,最大化模型在多样性测试样本上的性能,以此提高轨道缺陷检测系统的鲁棒性。结果显示相较于初训练模型,重训练模型在不同扰动下各方面性能都有所提升,重训练轨道缺陷检测模型在测试集上表现出更好的效果。
公开/授权文献
- CN117934819A 一种轨道缺陷检测系统鲁棒性提升方法 公开/授权日:2024-04-26