发明公开
- 专利标题: 一种基于多任务学习的车辆检索方法、装置、设备和介质
-
申请号: CN202310607631.9申请日: 2023-05-26
-
公开(公告)号: CN116501909A公开(公告)日: 2023-07-28
- 发明人: 王彦君 , 韩艺 , 郭彩丽 , 杨洋 , 刘芳芳
- 申请人: 中电信数智科技有限公司 , 北京邮电大学
- 申请人地址: 北京市海淀区复兴路33号13层东塔13层1308室;
- 专利权人: 中电信数智科技有限公司,北京邮电大学
- 当前专利权人: 中电信数智科技有限公司,北京邮电大学
- 当前专利权人地址: 北京市海淀区复兴路33号13层东塔13层1308室;
- 代理机构: 北京知汇林知识产权代理事务所
- 代理商 王俊杰
- 主分类号: G06F16/583
- IPC分类号: G06F16/583 ; G06V10/42 ; G06V10/44 ; G06V10/774 ; G06V10/82
摘要:
本申请提出一种基于多任务学习的车辆检索方法,包括如下步骤:获取包含车辆图像标注信息的训练数据集合、验证数据集合和测试数据集合,将所述训练数据集合的数据按照图像序号输入基于多任务学习的初始车辆检索模型中以训练得到车辆检索模型;将所述验证数据集合中的数据输入训练后的车辆检索模型,得到初步排序结果;采用多任务检索策略对所述初步排序结果进行重排序,得到最终排序结果。本申请通过多任务学习的方式指导车辆图像特征的提取,设计相机视角检索、车辆朝向检索作为辅助任务,消除检索时背景及视角不同带来的偏差,从而有效解决相机视角、车辆朝向变换导致的问题,以实现强可解释性、高效率、高检索精度的车辆检索。