unbco / solider Goto Github PK

Welcome to SOLIDER! SOLIDER is a Semantic Controllable Self-Supervised Learning Framework to learn general human representations from massive unlabeled human images which can benefit downstream human-centric tasks to the maximum extent. Unlike the existing self-supervised learning methods, prior knowledge from human images is utilized in SOLIDER to build pseudo semantic labels and import more semantic information into the learned representation. Meanwhile, different downstream tasks always require different ratios of semantic information and appearance information, and a single learned representation cannot fit for all requirements. To solve this problem, SOLIDER introduces a conditional network with a semantic controller, which can fit different needs of downstream tasks. For more details, please refer to our paper Beyond Appearance: a Semantic Controllable Self-Supervised Learning Framework for Human-Centric Visual Tasks.

• [2023/04/24: Codes of attribute recognition task is released!]
  • Training details of our pretrained model on downstream person attribute recognition task is released.
• [2023/03/28: Codes of 3 downstream tasks are released!]
  • Training details of our pretrained model on 3 downstream human visual tasks, including person re-identification, person search and pedestrian detection, are released.
• [2023/03/13: SOLIDER is accepted by CVPR2023!]
  • The paper of SOLIDER is accepted by CVPR2023, and its offical pytorch implementation is released in this repo.

This codebase has been developed with python version 3.7, PyTorch version 1.7.1, CUDA 10.1 and torchvision 0.8.2.

We use LUPerson as our training data, which consists of unlabeled human images. Download LUPerson from its offical link and unzip it.

• Choice 1. To train SOLIDER from scratch, please run:

sh run_solider.sh

• Choice 2. Training SOLIDER from scratch may take a long time. To speed up the training, you can train a DINO model first, and then finetune it with SOLIDER, as follows:

sh run_dino.sh
sh resume_solider.sh

There is a demo to run the trained SOLIDER model, which can be embedded into the inference or the downstream task finetuning.

python demo.py

We use Swin-Transformer as our backbone, which shows great advantages on many CV tasks.

• All the models are trained on the whole LUPerson dataset.

• Person Re-identification
• Person Search
• Pedestrian Detection
• Person Attribute Recognition
• Human Parsing
• Pose Estimation

Our implementation is mainly based on the following codebases. We gratefully thank the authors for their wonderful works.

• Swin-Transformer
• DINO
• TransReID
• TransReID-SSL
• SeqNet
• Pedestron
• LUPerson

If you use SOLIDER in your research, please cite our work by using the following BibTeX entry:

@inproceedings{chen2023beyond,
  title={Beyond Appearance: a Semantic Controllable Self-Supervised Learning Framework for Human-Centric Visual Tasks},
  author={Weihua Chen and Xianzhe Xu and Jian Jia and Hao Luo and Yaohua Wang and Fan Wang and Rong Jin and Xiuyu Sun},
  booktitle={The IEEE/CVF Conference on Computer Vision and Pattern Recognition},
  year={2023},
}