vamoko / pytorch-person-reid-baseline-pcb-beyond-part-models Goto Github PK

Pytorch-Person-ReID-PCB-Beyond-Part-Models

• A Strong Implementation of PCB (Beyond Part Models, ECCV2018) on Market-1501 and DukeMTMC-reID datasets.

• We support:

  • A strong PCB implementation which outperforms most existing implementations.
  • A simple and clear implementation, and end-to-end training and evaluation.

News

• We re-write a strong Re-ID baseline Bag of Tricks (BoT) with a more simple and clear implementation, which is more friendly with researchers and newers. Our code can be found here. BoT outperforms PCB by using only the global feature. Our implementation of BoT achieves the same performance with the offical one.

Dependencies

• Anaconda (Python 3.7)
• PyTorch 0.4.0
• GPU Memory >= 20G (we use 2 GTX1080ti)
• Memory >= 20G

Dataset Preparation

• Market-1501 Dataset and DukeMTMC-reID Dataset
• Download and extract both anywhere

Train and Test

python main.py --market_path market_path --duke_path duke_path

Experiments

1. Settings

• We conduct our experiments on 2 GTX1080ti GPUs

2. Results

Contacts

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

E-mail: [email protected]