发明公开
- 专利标题: 基于SAM模型的铁路轨道入侵数据标注和增强训练方法
-
申请号: CN202311845505.3申请日: 2023-12-29
-
公开(公告)号: CN118379615A公开(公告)日: 2024-07-23
- 发明人: 吉鹏程 , 张凌豪 , 李蒙 , 倪小龙 , 何潞 , 李健华 , 陈巧英 , 赵海涛 , 徐波
- 申请人: 南京泰通科技股份有限公司 , 南京邮电大学
- 申请人地址: 江苏省南京市雨花经济开发区凤汇大道35号;
- 专利权人: 南京泰通科技股份有限公司,南京邮电大学
- 当前专利权人: 南京泰通科技股份有限公司,南京邮电大学
- 当前专利权人地址: 江苏省南京市雨花经济开发区凤汇大道35号;
- 代理机构: 江苏圣典律师事务所
- 代理商 韩天宇
- 主分类号: G06V20/10
- IPC分类号: G06V20/10 ; G06V10/26 ; G06V10/25 ; G06V10/764 ; G06V10/774 ; G06V10/82 ; G06N3/0464 ; G06N3/09
摘要:
本发明提供了一种基于SAM模型的铁路轨道入侵数据标注和增强训练方法,以下步骤进行:1)异常入侵关键帧的提取。2)半自动标注过程:使用SAM‑RA模块的强大图像分割能力,对提取的关键帧进行标注,并生成入侵物的图像掩膜。3)数据增强:通过SAM‑RTN方法,使用生成的图像掩膜作为感知特征,进行特征提取和ROI采样。4)联合训练网络:将图像掩膜数据和感知特征的采样数据联合并输入到下游的目标检测和分类训练网络中。本发明的方法结合了SAM模型强大的图像分割能力和特征感知能力,从而显著提高了铁路轨道异常入侵物的目标检测和分类网络的性能。