摘要
本公开涉及计算机视觉技术领域,提供了一种行人重识别模型的训练方法、装置、电子设备及存储介质。该方法包括:将训练图像输入学生模型,对训练图像进行特征提取,并对特征提取得到的局部特征向量进行随机强度遮挡,得增强特征向量;将增强特征向量、第一初始类别特征向量输入学生模型的特征预测网络,得到第一特征向量和子图像的预测遮挡强度;将训练图像输入教师模型,得到第二特征向量;利用第二特征向量监督第一特征向量,利用随机遮挡强度监督子图像预测遮挡强度更新学生模型的参数和教师模型的参数,将训练完成的教师模型确定为行人重识别模型,提高了行人重识别模型在遮挡情况下的表现,增强行人重识别模型的鲁棒性和泛化能力。
技术关键词
行人重识别模型
图像
识别行人
学生
教师
模型更新
强度
训练集
注意力
参数
计算机视觉技术
语义
电子设备
可读存储介质
特征提取模块
处理器
非线性
传播算法
训练装置
系统为您推荐了相关专利信息
生成方法
实时视频流
节点
生成系统
图像特征信息
重载无人机
物资投放方法
输出特征
深度强化学习模型
融合特征
弱光图像增强方法
对比度
图像增强系统
迭代算法
参数