arnoldfychen / fast-reid

SOTA ReID Methods and Toolbox

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

FastReID

FastReID is a research platform that implements state-of-the-art re-identification algorithms. It is a groud-up rewrite of the previous verson, reid strong baseline.

What's New

  • Remove ignite(a high-level library) dependency and powered by PyTorch.
  • Includes more features such as circle loss, abundant visualization methods and evaluation metrics, SoTA results on conventional, cross-domain, partial and vehicle re-id, testing on multi-datasets simultaneously, etc.
  • Can be used as a library to support different projects on top of it. We'll open source more research projects in this way.
  • It trains much faster.

We write a chinese blog about this toolbox.

Installation

See INSTALL.md.

Quick Start

The designed architecture follows this guide PyTorch-Project-Template, you can check each folder's purpose by yourself.

See GETTING_STARTED.md.

Learn more at out documentation. And see projects/ for some projects that are build on top of fastreid.

Model Zoo and Baselines

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

Deployment

We provide some examples and scripts to convert fastreid model to Caffe, ONNX and TensorRT format in Fastreid deploy.

License

Fastreid is released under the Apache 2.0 license.

Citing Fastreid

If you use Fastreid in your research or wish to refer to the baseline results published in the Model Zoo, please use the following BibTeX entry.

@misc{he2020fastreid,
    title={FastReID: A Pytorch Toolbox for Real-world Person Re-identification},
    author={Lingxiao He and Xingyu Liao and Wu Liu and Xinchen Liu and Peng Cheng and Tao Mei},
    year={2020},
    eprint={2006.02631},
    archivePrefix={arXiv},
    primaryClass={cs.CV}
}

About

SOTA ReID Methods and Toolbox

License:Apache License 2.0


Languages

Language:Python 99.8%Language:Shell 0.1%Language:Makefile 0.0%