Torchreid: A Library for Deep Learning Person Re-Identification in Pytorch
This provides a software framework for researchers in computer vision to accelerate person re-identification research, but it is incremental as it builds on existing deep learning methods.
The authors tackled the problem of developing deep learning models for person re-identification by introducing Torchreid, a PyTorch-based library that supports 15 benchmark datasets and provides streamlined pipelines, resulting in a tool that facilitates fast development and evaluation of re-ID models.
Person re-identification (re-ID), which aims to re-identify people across different camera views, has been significantly advanced by deep learning in recent years, particularly with convolutional neural networks (CNNs). In this paper, we present Torchreid, a software library built on PyTorch that allows fast development and end-to-end training and evaluation of deep re-ID models. As a general-purpose framework for person re-ID research, Torchreid provides (1) unified data loaders that support 15 commonly used re-ID benchmark datasets covering both image and video domains, (2) streamlined pipelines for quick development and benchmarking of deep re-ID models, and (3) implementations of the latest re-ID CNN architectures along with their pre-trained models to facilitate reproducibility as well as future research. With a high-level modularity in its design, Torchreid offers a great flexibility to allow easy extension to new datasets, CNN models and loss functions.