一种基于预训练大模型的轨道缺陷检测方法及装置

    公开(公告)号:CN117455902A

    公开(公告)日:2024-01-26

    申请号:CN202311728678.7

    申请日:2023-12-15

    Abstract: 本发明涉及轨道缺陷检测技术领域,特别是指一种基于预训练大模型的轨道缺陷检测方法及装置,方法包括:基于图像随机掩码重建的预训练方法,对初始掩码自编码模型进行第一阶段预训练,得到第一中间训练模型;对第一中间训练模型进行第二阶段预训练,得到第二中间训练模型;基于ViTDet的训练方法,得到轨道重要部件位置检测模型;基于重要部件位置对第二中间训练模型进行训练,得到第三中间训练模型;根据第三中间训练模型以及第一中间训练模型进行数据自动增广,得到无病害样本以及有病害样本,根据样本对第三中间训练模型进行微调,得到训练好的轨道缺陷检测模型。采用本发明,可以解决高质量标注数据不足的问题,提高轨道缺陷检测的精度。

Patent Agency Ranking