摘要
本申请公开了一种模型训练方法、装置及目标装备的识别方法。其中,该方法包括:获取视频流中的多个图像帧;对图像帧进行卷积操作,并对卷积操作得到的特征进行批量归一化处理,得到第一目标特征图;利用不同尺度的卷积核对第一目标特征图进行卷积操作,并将不同尺度的卷积核的输出特征图进行融合,得到卷积层输出的n层的第二目标特征图;对n层的第二目标特征图中每层的第二目标特征图进行多尺度特征融合,得到每层的第二目标特征图对应的第一特征,对第一特征进行加权融合,得到第三目标特征图;利用第三目标特征图对深度学习模型进行训练,得到目标装备识别模型。本申请解决了相关目标装备识别模型的识别精度较低的技术问题。
技术关键词
模型训练方法
多尺度特征融合
装备
深度学习模型
非易失性存储介质
图像
输出特征
识别方法
批量
视频流
模型训练装置
对象
尺寸
计算机程序产品
坐标
参数
误差
网络
处理器
模块