# ReID_baseline
Baseline model (with bottleneck) for person ReID (using softmax and triplet loss). This is PyTorch version, [mxnet version](https://github.com/L1aoXingyu/reid_baseline_gluon) is here.

We support
- multi-GPU training
- easy dataset preparation
- end-to-end training and evaluation

## Get Started
1. `cd` to folder where you want to download this repo
2. Run `git clone https://github.com/L1aoXingyu/reid_baseline.git`
3. Install dependencies:
    - [pytorch 0.4](https://pytorch.org/)
    - torchvision
    - tensorflow (for tensorboard)
    - [tensorboardX](https://github.com/lanpa/tensorboardX)
4. Prepare dataset
    
    Create a directory to store reid datasets under this repo via
    ```bash
    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/
    ```
5. Prepare pretrained model if you don't have
    ```python
    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`

## Train
You can run 
```bash
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`.

## Results

| loss | rank1 | map |
| --- | --| ---|
| softmax | 87.9% | 70.1% |
| triplet | 88.8% | 74.8% | 
|triplet + softmax | 92.0% | 78.1% |