公开(公告)号：CN119810562A

公开(公告)日：2025-04-11

申请号：CN202510040872.9

申请日：2025-01-10

Applicant: 北京交通大学

Inventor： 阮涛 ,  郭保青 ,  王尧 ,  余祖俊 ,  朱力强 ,  李隽哲

IPC: G06V10/764 ,  G06V10/774 ,  G06V10/82 ,  G06V10/74 ,  G06N3/045 ,  G06N3/0464 ,  G06N3/084

Abstract: 本申请提供一种铁路入侵目标分类模型的训练方法、电子设备及存储介质，针对训练样本集中的每一铁路入侵目标类别，迭代以下步骤指定轮次以训练生成铁路入侵目标分类模型：通过待训练模型的特征提取层提取查询样本图像的第一特征提取结果和每张支持样本图像的第二特征提取结果；对第二特征提取结果进行修正，以获取第三特征提取结果；确定出支持样本图像属于实际铁路入侵目标类别的概率值；基于概率值和交叉熵损失函数进行反向传播，以更新所有铁路入侵目标类别对应的类中心所构成的类中心矩阵，用以在样本稀少分类类别开放总数不固定的条件下，训练生成识别精确的铁路入侵目标分类模型，并且模块具有对新分类类别的扩展性好。

公开(公告)号：CN119810452A

公开(公告)日：2025-04-11

申请号：CN202510040871.4

申请日：2025-01-10

Applicant: 北京交通大学

Inventor： 阮涛 ,  郭保青 ,  余祖俊 ,  朱力强 ,  王尧 ,  李隽哲

IPC: G06V10/26 ,  G06V10/82 ,  G06V20/70 ,  G06V10/74 ,  G06V10/764 ,  G06N3/045 ,  G06N3/0464 ,  G06N3/08

Abstract: 本申请提供一种基于循环一致的铁路入侵目标语义分割模型的训练方法，其中，方法包括通过特征提取网络分别提取查询样本图像的查询样本特征以及k个支持样本图像的k个支持样本特征；通过特征对齐模块对查询样本特征和多个支持样本特征分别进行自对齐处理，将自对齐后的查询样本特征和第k个支持样本特征进行交叉对齐；通过掩码匹配模块基于k个查询交叉特征和n个掩码建议，确定出支持原型；通过掩码匹配模块基于k个支持交叉特征和对应的标准掩码结果，确定出n个查询原型；通过掩码匹配模块基于支持原型、n个查询原型和n个掩码建议，确定出查询样本图像对应的预测掩码结果。