mirrors
/
fast-reid
mirror of https://github.com/JDAI-CV/fast-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
SOTA Re-identification Methods and Toolbox
111 Commits
1 Branch
4 Tags
15 MiB
Python 86.7%
C++ 11%
Cython 1.3%
CMake 0.6%
Dockerfile 0.4%
 
 
 
 
 
Go to file
liaoxingyu 6d96529d4c fix(data): fix resume training bug 
fix dataset pid dictionary loading bug when resume training,
data prefetcher will pre-load a batch of data, this will lead to
misalignment of old pid dict and updated pid dict.
We can address this problem by redefine a prefetcher in resume_or_load
 		2020-05-05 23:20:42 +08:00
configs chore(configs): update all training config 2020-05-01 09:04:51 +08:00
demo refactor bn_no_bias 2020-04-08 21:04:09 +08:00
fastreid fix(data): fix resume training bug 2020-05-05 23:20:42 +08:00
tests update version0.2 code 2020-03-25 10:58:26 +08:00
tools refactor(preciseBN): add preciseBN datasets show 2020-04-29 21:05:53 +08:00
.gitignore Change architecture: 2020-02-18 21:01:23 +08:00
MODEL_ZOO.md update model zoo 2020-05-04 14:36:16 +08:00
README.md style(configs): put all config files together 2020-04-29 16:18:54 +08:00

README.md

FastReID

FastReID is a research platform that implements state-of-the-art re-identification algorithms.

Quick Start

The designed architecture follows this guide 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

  3. Install dependencies:

  4. Prepare dataset Create a directory to store reid datasets under projects, for example

    cd fast-reid
mkdir datasets
    1. Download dataset to datasets/ from baidu pan or google driver
    2. Extract dataset. The dataset structure would like:
    datasets
    Market-1501-v15.09.15
        bounding_box_test/
        bounding_box_train/

  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. And then you can put it in ~/.cache/torch/checkpoints or anywhere you like.

    Then you should set the pretrain model path in configs/Base-bagtricks.yml.

  6. compile with cython to accelerate evalution

    cd fastreid/evaluation/rank_cylib; make all

Model Zoo and Baselines

We provide a large set of baseline results and trained models available for download in the Fastreid Model Zoo.