fast-reid/README.md

46 lines
1.6 KiB
Markdown
Raw Normal View History

# FastReID
2020-02-10 07:38:56 +08:00
FastReID is a research platform that implements state-of-the-art re-identification algorithms.
2020-02-10 07:38:56 +08:00
## Quick Start
2020-02-10 07:38:56 +08:00
The designed architecture follows this guide [PyTorch-Project-Template](https://github.com/L1aoXingyu/PyTorch-Project-Template), you can check each folder's purpose by yourself.
1. `cd` to folder where you want to download this repo
2. Run `git clone https://github.com/L1aoXingyu/fast-reid.git`
2020-02-10 07:38:56 +08:00
3. Install dependencies:
- [pytorch 1.0.0+](https://pytorch.org/)
- torchvision
- tensorboard
- [yacs](https://github.com/rbgirshick/yacs)
4. Prepare dataset
Create a directory to store reid datasets under projects, for example
2020-02-10 07:38:56 +08:00
```bash
cd fast-reid/projects/StrongBaseline
2020-02-10 07:38:56 +08:00
mkdir datasets
```
2020-02-10 07:38:56 +08:00
1. Download dataset to `datasets/` from [baidu pan](https://pan.baidu.com/s/1ntIi2Op) or [google driver](https://drive.google.com/file/d/0B8-rUzbwVRk0c054eEozWG9COHM/view)
2. Extract dataset. The dataset structure would like:
2020-02-10 07:38:56 +08:00
```bash
datasets
Market-1501-v15.09.15
bounding_box_test/
bounding_box_train/
```
2020-02-10 07:38:56 +08:00
5. Prepare pretrained model.
If you use origin ResNet, you do not need to do anything. But if you want to use ResNet_ibn, you need to download pretrain model in [here](https://drive.google.com/open?id=1thS2B8UOSBi_cJX6zRy6YYRwz_nVFI_S). And then you can put it in `~/.cache/torch/checkpoints` or anywhere you like.
Then you should set the pretrain model path in `configs/baseline_market1501.yml`.
2020-02-10 07:38:56 +08:00
6. compile with cython to accelerate evalution
2020-02-10 07:38:56 +08:00
```bash
cd fastreid/evaluation/rank_cylib; make all
2020-02-10 07:38:56 +08:00
```
## Model Zoo and Baselines