一种车辆属性识别模型的训练方法、系统及相关设备
基本信息
| 申请号 | CN202110827957.3 | 申请日 | - |
| 公开（公告）号 | CN113283549A | 公开（公告）日 | 2021-08-20 |
| 申请公布号 | CN113283549A | 申请公布日 | 2021-08-20 |
| 分类号 | G06K9/62（2006.01）I;G08G1/017（2006.01）I;G06N3/04（2006.01）I | 分类 | 计算；推算；计数; |
| 发明人 | 马永军;周东晖;闫潇宁 | 申请（专利权）人 | 深圳市安软慧视科技有限公司 |
| 代理机构 | 深圳君信诚知识产权代理事务所（普通合伙） | 代理人 | 刘伟 |
| 地址 | 518000广东省深圳市龙华新区龙华办事处清祥路清湖工业园宝能科技园7栋A座6楼ACDF | ||
| 法律状态 | - | ||
摘要
| 摘要 | 本发明提供了一种车辆属性识别模型的训练方法、系统及相关设备，训练方法包括以下步骤：构建包括可缩放的主干网络、BiFPN网络、跨类别注意力网络和属性分类网络的车辆属性识别模型；获取用于模型训练的图片集，其中图片包含车辆属性，并对图片进行数据增强，得到增强图片集；将增强图片集输入到主干网络中，得到初级特征图集；将初级特征图集利用BiFPN网络进行融合，得到融合特征图；将融合特征图输入到跨类别注意力网络，对其中图片的车辆属性进行权值标记和属性间信息提取，得到第三特征图集；将第三特征图集输入到属性分类网络，得到车辆属性识别结果。本发明使用跨类别的注意力机制构建车辆属性识别模型，提升了模型的泛化能力。 |
