Learning from Synthetic Data with Fine-grained Attributes for Person Re-Identification

Related tags

Deep LearningFineGPR
Overview

Less is More: Learning from Synthetic Data with Fine-grained Attributes for Person Re-Identification

Suncheng Xiang

Shanghai Jiao Tong University

Overview

In this paper, we construct and label a large-scale synthetic person dataset named FineGPR with fine-grained attribute distribution. Moreover, aiming to fully exploit the potential of FineGPR and promote the efficient training from millions of synthetic data, we propose an attribute analysis pipeline AOST to learn attribute distribution in target domain, then apply style transfer network to eliminate the gap between synthetic and real-world data and thus is freely deployed to new scenarios. Experiments conducted on benchmarks demonstrate that FineGPR with AOST outperforms (or is on par with) existing real and synthetic datasets, which suggests its feasibility for re-ID and proves the proverbial less-is-more principle. We hope this fine-grained dataset could advance research towards re-ID in real scenarios.


[Paper] [Video Sample] [Related Project]


πŸ”₯ NEWS πŸ”₯

  • [10/2021] πŸ“£ The first FineGPR-C caption dataset involving human describing event is coming !

  • [09/2021] πŸ“£ The large-scale synthetic person dataset FineGPR with fine-grained attribute distribution is released !


Table of Contents πŸ‘€


FineGPR Introduction

The FineGPR dataset is generated by a popular GTA5 game engine that can synthesise images under controllable viewpoints, weathers,illuminations and backgrounds, as well as 13 fine-grained attributes at the identity level πŸ‘ .

Our FineGPR dataset provides fine-grained and accurately configurable annotations, including 36 different viewpoints, 7 different kinds of weathers, 7 different kinds of illuminations, and 9 different kinds of backgrounds.

Viewpoint πŸ“·

Definition of different viewpoints. Viewpoints of one identity are sampled at an interval of 10Β°, e.g. 0Β°-80Β° denotes that a person has 9 different angles in total.

Weather 🌨 and Illumination πŸŽ‡

The exemplars of different weather distribution (left) and illumination distribution (right) from the proposed FineGPR dataset.

Attributes at the Identity Level ⛹️‍♀️

The distributions of attributes at the identity level on FineGPR. The left figure shows the numbers of IDs for each attribute. The middle and right pies illustrate the distribution of the colors of upper-body and low-body clothes respectively.

Some visual exemplars with ID-level pedestrian attributes in the proposed FineGPR dataset, such as Wear short sleeve , Wear dress, Wear hat, Carry bag, etc.


Comparison with existing datasets

Some Mainstream Datasets for Person Re-Identification

For related FineGPR dataset (details of the previous related work, please refer to the our homepage GPR πŸ”Ž :

dataset IDs (ID-Attributes) boxs cams weathers illumination scene resolution
Market-1501 1,501 ( βœ”οΈ ) 32,668 6 - - - low
CUHK03 1,467 ( ❌ ) 14,096 2 - - - low
DukeMTMC-reID 1,404 ( βœ”οΈ ) 36,411 8 - - - low
MSMT17 4,101 ( ❌ ) 126,441 15 - - - vary
SOMAset 50 ( ❌ ) 100,000 250 - - - -
SyRI 100 ( ❌ ) 1,680,000 100 - 140 - -
PersonX 1,266 ( ❌ ) 273,456 6 - - 1 vary
Unreal 3,000 ( ❌ ) 120,000 34 - - 1 low
RandPerson 8,000 ( ❌ ) 1,801,816 19 - - 4 low
FineGPR 1150 ( βœ”οΈ ) 2,028,600 36 7 7 9 high

Link of the Dataset

Data of FineGPR for Viewpoint Analysis

A small subset of FineGPR can be downloaded from the following links:

Directories & Files of images

FineGPR_Dataset 
β”œβ”€β”€ FineGPR/   # This file is our original dataset, we provide the samples of ID=0001 and ID=0003 in this file folder.
β”‚   β”œβ”€β”€ 0001
β”‚   β”‚   β”œβ”€β”€ 0001_c01_w01_l01_p01.jpg 
β”‚   β”‚	β”œβ”€β”€ 0001_c01_w01_l02_p01.jpg  
β”‚   β”‚   β”œβ”€β”€ 0001_c01_w01_l03_p01.jpg
β”‚   β”‚   └── ...
β”‚   β”œβ”€β”€ 0003/
β”‚   β”‚   β”œβ”€β”€ 0003_c01_w01_l01_p06.jpg  
β”‚   β”‚   β”œβ”€β”€ 0003_c01_w01_l02_p06.jpg
β”‚   β”‚   β”œβ”€β”€ 0003_c01_w01_l03_p06.jpg	   
β”‚   β”‚   └── ...
β”‚   └── ...
β”œβ”€β”€ FineGPR_subset   # This file is the subset of FineGPR dataset, each Identity contains 4 images. 
β”‚   β”œβ”€β”€ 0001_c01_w03_l05_p03.jpg 
β”‚   β”œβ”€β”€ 0001_c10_w03_l05_p03.jpg
β”‚   β”œβ”€β”€ 0001_c19_w03_l05_p03.jpg
β”‚   β”œβ”€β”€ 0001_c28_w03_l05_p03.jpg
β”‚   β”œβ”€β”€ 0003_c01_w03_l05_p08.jpg 
β”‚   β”œβ”€β”€ 0003_c10_w03_l05_p08.jpg
β”‚   β”œβ”€β”€ 0003_c19_w03_l05_p08.jpg
β”‚   β”œβ”€β”€ 0003_c28_w03_l05_p08.jpg  
β”‚   └── ...
└── README.md   # Readme file

Name of the image

Taking "0001_c01_w01_l01_p01.jpg" as an example:

  • 0001 is the id of the person
  • c01 is the id of the camera
  • w01 is the id of the weather
  • l01 is the id of the illumination
  • p01 is the id of the background

Viewpoint annotations

FineGPR
β”œβ”€β”€ c01:90Β°      β”œβ”€β”€ c10:180Β°      β”œβ”€β”€ c19:270Β°      β”œβ”€β”€ c28:0Β°
β”œβ”€β”€ c02:100Β°     β”œβ”€β”€ c11:190Β°      β”œβ”€β”€ c20:280Β°      β”œβ”€β”€ c29:10Β°
β”œβ”€β”€ c03:110Β°     β”œβ”€β”€ c12:200Β°      β”œβ”€β”€ c21:290Β°      β”œβ”€β”€ c30:20Β°
β”œβ”€β”€ c04:120Β°     β”œβ”€β”€ c13:210Β°      β”œβ”€β”€ c22:300Β°      β”œβ”€β”€ c31:30Β°
β”œβ”€β”€ c05:130Β°     β”œβ”€β”€ c14:220Β°      β”œβ”€β”€ c23:310Β°      β”œβ”€β”€ c32:40Β°
β”œβ”€β”€ c06:140Β°     β”œβ”€β”€ c15:230Β°      β”œβ”€β”€ c24:320Β°      β”œβ”€β”€ c33:50Β°
β”œβ”€β”€ c07:150Β°     β”œβ”€β”€ c16:240Β°      β”œβ”€β”€ c25:330Β°      β”œβ”€β”€ c34:60Β°
β”œβ”€β”€ c08:160Β°     β”œβ”€β”€ c17:250Β°      β”œβ”€β”€ c26:340Β°      β”œβ”€β”€ c35:70Β°
└── c09:170Β°     └── c18:260Β°      └── c27:350Β°      └── c36:80Β°

Weather annotations

FineGPR
β”œβ”€β”€ w01:Sunny
β”œβ”€β”€ w02:Clouds    
β”œβ”€β”€ w03:Overcast
β”œβ”€β”€ w04:Foggy   
β”œβ”€β”€ w05:Neutral
β”œβ”€β”€ w06:Blizzard 
└── w07:Snowlight 	   

Illumination annotations

FineGPR
β”œβ”€β”€ l01:Midnight
β”œβ”€β”€ l02:Dawn    
β”œβ”€β”€ l03:Forenoon
β”œβ”€β”€ l04:Noon   
β”œβ”€β”€ l05:Afternoon
β”œβ”€β”€ l06:Dusk 
└── l07:Night 	   

Scene annotations

FineGPR
β”œβ”€β”€ p01:Urban
β”œβ”€β”€ p02:Urban   
β”œβ”€β”€ p03:Wild
β”œβ”€β”€ p04:Urban   
β”œβ”€β”€ p05:Wild
β”œβ”€β”€ p06:Urban
β”œβ”€β”€ p07:Urban
β”œβ”€β”€ p08:Wild 
└── p09:Urban 	   

Method

πŸ’‘ The two-stage pipeline AOST to learn attribute distribution of target domain. Firstly, we learn attribute distribution of real domain on the basis of XGBoost & PSO learning system. Secondly, we perform style transfer to enhance the reality of optimal dataset. Finally, the transferred data are adopted for downstream re-ID task.


Results

Performance comparison with existing Real and Synthetic datasets on Market-1501, DukeMTMC-reID and CUHK03, respectively.

References

  • [1] Image-image domain adaptation with preserved self-similarity and domain-dissimilarity for person re-identification. CVPR 2018.
  • [2] Bag of tricks and a strong baseline for deep person re-identification. CVPRW 2019.

Extendibility

Accompanied with our FineGPR, we also provide some human body masks (Middle) and keypoint locations (Bottom) of all characters during the annotation. We hope that our synthetic dataset FineGPR can not only contribute a lot to the development of generalizable person re-ID, but also advance the research of other computer vision tasks, such as human part segmentation and pose estimation.

FineGPR-C caption dataset

On the basis of FineGPR dafaset, we introduce a dynamic strategy to generate high-quality captions with fine-grained attribute annotations for semantic-based pretraining. To be more specific, we rearrange the different attributes as word embeddings into caption formula in the different position, and then generate semantically dense caption with high-quality description, which gives rise to our newly constructed FineGPR-C caption dataset.

A small subset of FineGPR-C caption dataset can be downloaded from the following links:

Citation

If you use our FineGPR dataset for your research, please cite our Paper.

@article{xiang2021less,
  title={Less is More: Learning from Synthetic Data with Fine-grained Attributes for Person Re-Identification},
  author={Xiang, Suncheng and You, Guanjie and Guan, Mengyuan and Chen, Hao and Wang, Feng and Liu, Ting and Fu, Yuzhuo},
  journal={arXiv preprint arXiv:2109.10498},
  year={2021}
}

If you do think this FineGPR-C caption dataset is useful and have used it in your research, please cite our Paper.

@article{xiang2021vtbr,
  title={VTBR: Semantic-based Pretraining for Person Re-Identification},
  author={Xiang, Suncheng and Zhang, Zirui and Guan, Mengyuan and Chen, Hao and Yan, Binjie and Liu, Ting and Fu, Yuzhuo},
  journal={arXiv preprint arXiv:2110.05074},
  year={2021}
}

Ethical Considerations

Our task and dataset were created with careful attention to ethical questions, which we encountered throughout our work. Access to our dataset will be provided for research purposes only and with restrictions on redistribution. Additionally, as we filtered out the sensitive attribute name in our fine-grained attribute annotation, our dataset cannot be easily repurposed for unintended tasks. Importantly, we are very cautious of human-annotation procedure of large scale datasets towards the social and ethical implications. Furthermore, we do not consider the datasets for developing non-research systems without further processing or augmentation. We hope this fine-grained dataset will shed light into potential tasks for the research community to move forward.


LICENSE

  • The FineGPR Dataset and FineGPR-C caption is made available for non-commercial purposes only.
  • You will not, directly or indirectly, reproduce, use, or convey the FineGPR dataset and FineGPR-C caption dataset or any Content, or any work product or data derived therefrom, for commercial purposes.

Permissions of this strong copyleft license (GNU General Public License v3.0) are conditioned on making available complete source code of licensed works and modifications, which include larger works using a licensed work, under the same license. Copyright and license notices must be preserved. Contributors provide an express grant of patent rights.


Acknowledgements

This research was supported by the National Natural Science Foundation of China under Project (Grant No. 61977045). We would like to thank authors of FineGPR, and FineGPR-Caption dataset for their work. They provide tremendous efforts in these dataset to advance the research in this field. We also appreciate Zefang Yu, Mingye Xie and Guanjie You for insightful feedback and discussion.


For further questions and suggestions about our datasets and methods, please feel free to contact Suncheng Xiang: [email protected]

Owner
SunchengXiang
SunchengXiang
Code for the head detector (HeadHunter) proposed in our CVPR 2021 paper Tracking Pedestrian Heads in Dense Crowd.

Head Detector Code for the head detector (HeadHunter) proposed in our CVPR 2021 paper Tracking Pedestrian Heads in Dense Crowd. The head_detection mod

Ramana Sundararaman 76 Dec 06, 2022
Official implementation of the MM'21 paper Constrained Graphic Layout Generation via Latent Optimization

[MM'21] Constrained Graphic Layout Generation via Latent Optimization This repository provides the official code for the paper "Constrained Graphic La

Kotaro Kikuchi 73 Dec 27, 2022
Python tools for 3D face: 3DMM, Mesh processing(transform, camera, light, render), 3D face representations.

face3d: Python tools for processing 3D face Introduction This project implements some basic functions related to 3D faces. You can use this to process

Yao Feng 2.3k Dec 30, 2022
Toward Spatially Unbiased Generative Models (ICCV 2021)

Toward Spatially Unbiased Generative Models Implementation of Toward Spatially Unbiased Generative Models (ICCV 2021) Overview Recent image generation

Jooyoung Choi 88 Dec 01, 2022
GemNet model in PyTorch, as proposed in "GemNet: Universal Directional Graph Neural Networks for Molecules" (NeurIPS 2021)

GemNet: Universal Directional Graph Neural Networks for Molecules Reference implementation in PyTorch of the geometric message passing neural network

Data Analytics and Machine Learning Group 124 Dec 30, 2022
Mmdetection3d Noted - MMDetection3D is an open source object detection toolbox based on PyTorch

MMDetection3D is an open source object detection toolbox based on PyTorch

Jiangjingwen 13 Jan 06, 2023
68 keypoint annotations for COFW test data

68 keypoint annotations for COFW test data This repository contains manually annotated 68 keypoints for COFW test data (original annotation of CFOW da

31 Dec 06, 2022
[ICML 2020] Prediction-Guided Multi-Objective Reinforcement Learning for Continuous Robot Control

PG-MORL This repository contains the implementation for the paper Prediction-Guided Multi-Objective Reinforcement Learning for Continuous Robot Contro

MIT Graphics Group 65 Jan 07, 2023
Neural Radiance Fields Using PyTorch

This project is a PyTorch implementation of Neural Radiance Fields (NeRF) for reproduction of results whilst running at a faster speed.

Vedant Ghodke 1 Feb 11, 2022
CoINN: Correlated-informed neural networks: a new machine learning framework to predict pressure drop in micro-channels

CoINN: Correlated-informed neural networks: a new machine learning framework to predict pressure drop in micro-channels Accurate pressure drop estimat

Alejandro Montanez 0 Jan 21, 2022
This repository contains a PyTorch implementation of "AD-NeRF: Audio Driven Neural Radiance Fields for Talking Head Synthesis".

AD-NeRF: Audio Driven Neural Radiance Fields for Talking Head Synthesis | Project Page | Paper | PyTorch implementation for the paper "AD-NeRF: Audio

551 Dec 29, 2022
Implementation for Paper "Inverting Generative Adversarial Renderer for Face Reconstruction"

StyleGAR TODO: add arxiv link Implementation of Inverting Generative Adversarial Renderer for Face Reconstruction TODO: for test Currently, some model

155 Oct 27, 2022
ROSITA: Enhancing Vision-and-Language Semantic Alignments via Cross- and Intra-modal Knowledge Integration

ROSITA News & Updates (24/08/2021) Release the demo to perform fine-grained semantic alignments using the pretrained ROSITA model. (15/08/2021) Releas

Vision and Language Group@ MIL 48 Dec 23, 2022
Pytorch code for "Text-Independent Speaker Verification Using 3D Convolutional Neural Networks".

:speaker: Deep Learning & 3D Convolutional Neural Networks for Speaker Verification

Amirsina Torfi 114 Dec 18, 2022
for taichi voxel-challange event

Taichi Voxel Challenge Figure: result of python3 example6.py. Please replace the image above (demo.jpg) with yours, so that other people can immediate

Liming Xu 20 Nov 26, 2022
PyTorch implementation of the YOLO (You Only Look Once) v2

PyTorch implementation of the YOLO (You Only Look Once) v2 The YOLOv2 is one of the most popular one-stage object detector. This project adopts PyTorc

η”³η‘žη‰ (Ruimin Shen) 433 Nov 24, 2022
Our VMAgent is a platform for exploiting Reinforcement Learning (RL) on Virtual Machine (VM) scheduling tasks.

VMAgent is a platform for exploiting Reinforcement Learning (RL) on Virtual Machine (VM) scheduling tasks. VMAgent is constructed based on one month r

56 Dec 12, 2022
HDMapNet: A Local Semantic Map Learning and Evaluation Framework

HDMapNet_devkit Devkit for HDMapNet. HDMapNet: A Local Semantic Map Learning and Evaluation Framework Qi Li, Yue Wang, Yilun Wang, Hang Zhao [Paper] [

Tsinghua MARS Lab 421 Jan 04, 2023
Simple-Neural-Network From Scratch in Python

Simple-Neural-Network From Scratch in Python This is a simple Neural Network created without any Machine Learning Libraries. The only dependencies are

Aum Shah 1 Dec 28, 2021
ICRA 2021 "Towards Precise and Efficient Image Guided Depth Completion"

PENet: Precise and Efficient Depth Completion This repo is the PyTorch implementation of our paper to appear in ICRA2021 on "Towards Precise and Effic

232 Dec 25, 2022