SD-ReID: View-aware Stable Diffusion for Aerial-Ground Person Re-Identification

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Wang, Yuhao, Hu, Xiang, Wang, Lixin, Zhang, Pingping, Lu, Huchuan
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866917492327186432
author Wang, Yuhao
Hu, Xiang
Wang, Lixin
Zhang, Pingping
Lu, Huchuan
author_facet Wang, Yuhao
Hu, Xiang
Wang, Lixin
Zhang, Pingping
Lu, Huchuan
contents Aerial-Ground Person Re-IDentification (AG-ReID) aims to retrieve specific persons across cameras with different viewpoints. Previous works focus on designing discriminative models to maintain the identity consistency despite drastic changes in camera viewpoints. The core idea behind these methods is quite natural, but designing a view-robust model is a very challenging task. Moreover, they overlook the contribution of view-specific features in enhancing the model's ability to represent persons. To address these issues, we propose a novel generative framework named SD-ReID for AG-ReID, which leverages generative models to mimic the feature distribution of different views while extracting robust identity representations. More specifically, we first train a ViT-based model to extract person representations along with controllable conditions, including identity and view conditions. We then fine-tune the Stable Diffusion (SD) model to enhance person representations guided by these controllable conditions. Furthermore, we introduce the View-Refined Decoder (VRD) to bridge the gap between instance-level and global-level features. Finally, both person representations and all-view features are employed to retrieve target persons. Extensive experiments on five AG-ReID benchmarks (i.e., CARGO, AG-ReIDv1, AG-ReIDv2, LAGPeR and G2APS-ReID) demonstrate the effectiveness of our proposed method. The source code and pre-trained models are available at https://github.com/924973292/SD-ReID.
format Preprint
id arxiv_https___arxiv_org_abs_2504_09549
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle SD-ReID: View-aware Stable Diffusion for Aerial-Ground Person Re-Identification
Wang, Yuhao
Hu, Xiang
Wang, Lixin
Zhang, Pingping
Lu, Huchuan
Computer Vision and Pattern Recognition
Aerial-Ground Person Re-IDentification (AG-ReID) aims to retrieve specific persons across cameras with different viewpoints. Previous works focus on designing discriminative models to maintain the identity consistency despite drastic changes in camera viewpoints. The core idea behind these methods is quite natural, but designing a view-robust model is a very challenging task. Moreover, they overlook the contribution of view-specific features in enhancing the model's ability to represent persons. To address these issues, we propose a novel generative framework named SD-ReID for AG-ReID, which leverages generative models to mimic the feature distribution of different views while extracting robust identity representations. More specifically, we first train a ViT-based model to extract person representations along with controllable conditions, including identity and view conditions. We then fine-tune the Stable Diffusion (SD) model to enhance person representations guided by these controllable conditions. Furthermore, we introduce the View-Refined Decoder (VRD) to bridge the gap between instance-level and global-level features. Finally, both person representations and all-view features are employed to retrieve target persons. Extensive experiments on five AG-ReID benchmarks (i.e., CARGO, AG-ReIDv1, AG-ReIDv2, LAGPeR and G2APS-ReID) demonstrate the effectiveness of our proposed method. The source code and pre-trained models are available at https://github.com/924973292/SD-ReID.
title SD-ReID: View-aware Stable Diffusion for Aerial-Ground Person Re-Identification
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2504.09549