Skip to content

wangguanan/light-reid

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

61 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Pytorch-Person-ReID-Baseline-Bag-of-Tricks

  • Introduction: This project re-implements the strong person re-identification baseline: Bag of Tricks (paper and official code).
  • Advantage: This project provides a more simple and clear implementation by only using the best parameters and removing lots of unnecessary modules.

News

  • 2020-03-27: [CVPR'20] Our new work about Occluded ReID has been accepted by CVPR'20. (Paper)
  • 2020-01-01: [AAAI'20] Our new work about RGB-Infrared(IR) ReID for dark situation has been accepted by AAAI'20. (Paper, Code).
  • 2019-10-25: [ICCV'19] Our new work about RGB-Infrared(IR) ReID for dark situation has been accepted by ICCV'19. (Paper, Code).
  • 2019-05-01: We implement PCB and achieve better performance than the offical one. (Code)

Update

  • 2020-03-27: this branch is will not be updated, which based on python2.7 and torch-0.4.0. Please find new updating on the master branch, which is based on python3.7 and torch-1.1.0.
  • 2019-06-18: we add visualization code to show ranked images

Dependencies

Dataset Preparation

Run

# train
python main.py --market_path market_path --duke_path duke_path --output_path output_path/ --mode train 
# test, the output_path should be same with that in training process
python main.py --market_path market_path --duke_path duke_path --output_path output_path/ --mode test --resume_test_epoch resume_test_epoch
# visualize the ranked images, the output_path should be same with that in training process
python main.py --market_path market_path --duke_path duke_path --output_path output_path/ --mode visualize --resume_visualize_epoch resume_visualize_epoch

Experiments

1. Tricks we used

  • Warm up learning rate
  • Random erasing augmentation (REA)
  • Label smoothing
  • Last stride
  • BNNeck
  • Note that our implementation uses no the center loss and re-ranking.

2. Settings

  • We conduct our experiments on 1 GTX1080ti GPU

3. Results (with REA)

Repeat market2market market2duke duke2duke duke2market
1 0.939 (0.858) 0.290 (0.159) 0.874 (0.767) 0.486 (0.210)
2 0.944 (0.858) 0.295 (0.156) 0.868 (0.765) 0.492 (0.223)
3 0.942 (0.859) 0.281 (0.152) 0.863 (0.765) 0.485 (0.221)
Average 0.942 (0.858) 0.289 (0.156) 0.868 (0.766) 0.488 (0.218)
Paper 0.941 (0.857) - 0.864 (0.764)

4. Results (without REA)

Repeat market2market market2duke duke2duke duke2market
1 0.936 (0.824) 0.427 (0.264) 0.849 (0.714) 0.556 (0.269)
Paper - 0.414(0.257) - 0.543 (0.255)

5. Visualization of Ranked Images on Market-1501 Dataset (with REA)

Query Top1 Top2 Top3 Top4 Top5 Top6 Top7 Top8 Top9 Top10
  • More results can be seen in folder ranked_images/market

Contacts

If you have any question about the project, please feel free to contact with me.

E-mail: guan.wang0706@gmail.com

About

[ECCV2020] a toolbox of light-reid learning for faster inference, speed both feature extraction and retrieval stages up to >30x

Topics

Resources

Stars

Watchers

Forks

Packages

 
 
 

Contributors

Languages