基于多尺度掩码特征自编码器的城轨车底异常检测方法

    公开(公告)号:CN116612351A

    公开(公告)日:2023-08-18

    申请号:CN202310596242.0

    申请日:2023-05-24

    发明人: 王红军 陈云

    摘要: 本发明公开了一种基于多尺度掩码特征自编码器的城轨车底异常检测方法,涉及轨道交通技术领域,所述方法包括基于车底监测数据构建训练集,所述训练集包括大规模无标签训练集和少量有标签训练集;构建多尺度掩码特征自编码器,利用所述无标签训练集对所述多尺度掩码特征自编码器进行重建图像的自监督训练,得到模型参数;将获取的多尺度掩码特征自编码器中编码器部分的模型参数和相应的网络结构作为骨架网络嵌入到下游车底异常检测任务的网络结构中,并利用少量有标签训练集对模型参数进行有监督的微调训练,得到车底异常识别模型。本发明可以有效地解决现有技术中数据标注困难、元数据信息利用率低、正负样本不均匀以及受光学成像环境干扰等问题。