ReID_baseline

Baseline model (with bottleneck) for person ReID (using softmax and triplet loss). This is PyTorch version, mxnet version is here.

We support

Get Started

cd to folder where you want to download this repo
Run git clone https://github.com/L1aoXingyu/reid_baseline.git
Install dependencies:
- pytorch 0.4
- torchvision
- tensorflow (for tensorboard)
- tensorboardX
Prepare dataset

Create a directory to store reid datasets under this repo via
```
cd reid_baseline
mkdir data
```
1. Download dataset to data/ from http://www.liangzheng.org/Project/project_reid.html
2. Extract dataset and rename to market1501. The data structure would like:
```
market1501/
    bounding_box_test/
    bounding_box_train/
```
Prepare pretrained model if you don't have
```
from torchvision import models
models.resnet50(pretrained=True)
```
Then it will automatically download model in ~.torch/models/, you should set this path in config.py

You can run

bash scripts/train_triplet_softmax.sh

in reid_baseline folder if you want to train with softmax and triplet loss. You can find others train scripts in scripts.