search

Cysu / open-reid

Open source person re-identification library in python
https://cysu.github.io/open-reid/
MIT License
1.34k stars 350 forks source link
deep-learning metric-learning person-reidentification python pytorch

readme

Open-ReID

Open-ReID is a lightweight library of person re-identification for research purpose. It aims to provide a uniform interface for different datasets, a full set of models and evaluation metrics, as well as examples to reproduce (near) state-of-the-art results.

Installation

Install PyTorch (version >= 0.2.0). Although we support both python2 and python3, we recommend python3 for better performance.

git clone https://github.com/Cysu/open-reid.git
cd open-reid
python setup.py install

Examples

python examples/softmax_loss.py -d viper -b 64 -j 2 -a resnet50 --logs-dir logs/softmax-loss/viper-resnet50

This is just a quick example. VIPeR dataset may not be large enough to train a deep neural network.

Check about more examples and benchmarks.