The Solution for the CVPR 2023 1st foundation model challenge-Track2

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Xu, Haonan, Huang, Yurui, Pan, Sishun, Guan, Zhihao, Xu, Yi, Yang, Yang
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866911823202091008
author Xu, Haonan
Huang, Yurui
Pan, Sishun
Guan, Zhihao
Xu, Yi
Yang, Yang
author_facet Xu, Haonan
Huang, Yurui
Pan, Sishun
Guan, Zhihao
Xu, Yi
Yang, Yang
contents In this paper, we propose a solution for cross-modal transportation retrieval. Due to the cross-domain problem of traffic images, we divide the problem into two sub-tasks of pedestrian retrieval and vehicle retrieval through a simple strategy. In pedestrian retrieval tasks, we use IRRA as the base model and specifically design an Attribute Classification to mine the knowledge implied by attribute labels. More importantly, We use the strategy of Inclusion Relation Matching to make the image-text pairs with inclusion relation have similar representation in the feature space. For the vehicle retrieval task, we use BLIP as the base model. Since aligning the color attributes of vehicles is challenging, we introduce attribute-based object detection techniques to add color patch blocks to vehicle images for color data augmentation. This serves as strong prior information, helping the model perform the image-text alignment. At the same time, we incorporate labeled attributes into the image-text alignment loss to learn fine-grained alignment and prevent similar images and texts from being incorrectly separated. Our approach ranked first in the final B-board test with a score of 70.9.
format Preprint
id arxiv_https___arxiv_org_abs_2403_17702
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle The Solution for the CVPR 2023 1st foundation model challenge-Track2
Xu, Haonan
Huang, Yurui
Pan, Sishun
Guan, Zhihao
Xu, Yi
Yang, Yang
Computer Vision and Pattern Recognition
In this paper, we propose a solution for cross-modal transportation retrieval. Due to the cross-domain problem of traffic images, we divide the problem into two sub-tasks of pedestrian retrieval and vehicle retrieval through a simple strategy. In pedestrian retrieval tasks, we use IRRA as the base model and specifically design an Attribute Classification to mine the knowledge implied by attribute labels. More importantly, We use the strategy of Inclusion Relation Matching to make the image-text pairs with inclusion relation have similar representation in the feature space. For the vehicle retrieval task, we use BLIP as the base model. Since aligning the color attributes of vehicles is challenging, we introduce attribute-based object detection techniques to add color patch blocks to vehicle images for color data augmentation. This serves as strong prior information, helping the model perform the image-text alignment. At the same time, we incorporate labeled attributes into the image-text alignment loss to learn fine-grained alignment and prevent similar images and texts from being incorrectly separated. Our approach ranked first in the final B-board test with a score of 70.9.
title The Solution for the CVPR 2023 1st foundation model challenge-Track2
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2403.17702