一种车辆重识别模型的训练方法、系统及相关设备

基本信息

申请号 CN202110723839.8 申请日 -
公开(公告)号 CN113255601A 公开(公告)日 2021-08-13
申请公布号 CN113255601A 申请公布日 2021-08-13
分类号 G06K9/00(2006.01)I;G06K9/62(2006.01)I;G06N3/04(2006.01)I;G06N3/08(2006.01)I 分类 计算;推算;计数;
发明人 王嫄;曾磊磊;王广义;熊宁;杨巨成;闫潇宁 申请(专利权)人 深圳市安软慧视科技有限公司
代理机构 深圳君信诚知识产权代理事务所(普通合伙) 代理人 刘伟
地址 518000广东省深圳市龙华新区龙华办事处清祥路清湖工业园宝能科技园7栋A座6楼ACDF
法律状态 -

摘要

摘要 本发明适用于计算机视觉领域,提供了一种车辆重识别模型的训练方法、系统及相关设备,所述训练方法包括以下步骤:对数据集的图片进行筛选和分类,获取具有车辆特征的训练集、验证集;构建一种平行堆叠且拓扑结构相同的分组卷积块;利用所述分组卷积块构建基于分组卷积深度残差网络的车辆重识别模型;获取预训练数据集,并对所述车辆重识别模型进行预训练;获取微调数据集和半监督数据集,并对所述车辆重识别模型进行微调;利用所述数据集和所述验证集对所述车辆重识别模型进行测试,获取最终的所述车辆重识别模型。本发明在深度残差网络的基础之上进一步使用了分组卷积,降低了复杂度,提高了车辆重识别的性能。