一种基于多任务学习的车辆检索方法、装置、设备和介质
摘要:
本申请提出一种基于多任务学习的车辆检索方法,包括如下步骤:获取包含车辆图像标注信息的训练数据集合、验证数据集合和测试数据集合,将所述训练数据集合的数据按照图像序号输入基于多任务学习的初始车辆检索模型中以训练得到车辆检索模型;将所述验证数据集合中的数据输入训练后的车辆检索模型,得到初步排序结果;采用多任务检索策略对所述初步排序结果进行重排序,得到最终排序结果。本申请通过多任务学习的方式指导车辆图像特征的提取,设计相机视角检索、车辆朝向检索作为辅助任务,消除检索时背景及视角不同带来的偏差,从而有效解决相机视角、车辆朝向变换导致的问题,以实现强可解释性、高效率、高检索精度的车辆检索。
0/0