Voltar ao ranking

Cysu/open-reid

Pythoncysu.github.io/open-reid/

Open source person re-identification library in python

person-reidentificationdeep-learningmetric-learningpytorchpython
Crescimento de estrelas
Estrelas
1.4k
Forks
350
Crescimento semanal
Issues
36
5001k
mar. de 2017abr. de 2020jun. de 2023jul. de 2026
ArtefatosPyPIpip install open-reid
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.