摘要
本申请提供一种铁路入侵目标分类模型的训练方法、电子设备及存储介质,针对训练样本集中的每一铁路入侵目标类别,迭代以下步骤指定轮次以训练生成铁路入侵目标分类模型:通过待训练模型的特征提取层提取查询样本图像的第一特征提取结果和每张支持样本图像的第二特征提取结果;对第二特征提取结果进行修正,以获取第三特征提取结果;确定出支持样本图像属于实际铁路入侵目标类别的概率值;基于概率值和交叉熵损失函数进行反向传播,以更新所有铁路入侵目标类别对应的类中心所构成的类中心矩阵,用以在样本稀少分类类别开放总数不固定的条件下,训练生成识别精确的铁路入侵目标分类模型,并且模块具有对新分类类别的扩展性好。
技术关键词
残差模块
铁路
图像
矩阵
机器可读指令
电子设备
损失函数优化
处理器
可读存储介质
查询特征
训练样本集
神经网络模型
存储器
算法
计算机
特征值
参数
系统为您推荐了相关专利信息
矿区土壤重金属
无人机遥感影像
反演方法
卫星遥感影像
图像
表面缺陷检测方法
幻影
钢铁
工业图像处理技术
注意力机制
缺陷检测方法
锡球
深度特征融合
BGA封装芯片
深度学习网络