Torchreid: A Library for Deep Learning Person Re-Identification in Pytorch

Abstract

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.

arXiv commentTech report

References in corpus (14)

• Adam: A Method for Stochastic Optimization
• Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift
• TensorFlow: Large-Scale Machine Learning on Heterogeneous Distributed Systems
• HuggingFace's Transformers: State-of-the-art Natural Language Processing
• In Defense of the Triplet Loss for Person Re-Identification
• MXNet: A Flexible and Efficient Machine Learning Library for Heterogeneous Distributed Systems
• On the Convergence of Adam and Beyond
• Paying More Attention to Attention: Improving the Performance of Convolutional Neural Networks via Attention Transfer
• MMDetection: Open MMLab Detection Toolbox and Benchmark
• Deep Transfer Learning for Person Re-identification
• The Devil is in the Middle: Exploiting Mid-level Representations for Cross-Domain Instance Matching
• Learning Generalisable Omni-Scale Representations for Person Re-Identification
• Torchmeta: A Meta-Learning library for PyTorch
• SimpleDet: A Simple and Versatile Distributed Framework for Object Detection and Instance Recognition

Cited by in corpus (18)

• Omni-Scale Feature Learning for Person Re-Identification
• DeepSportradar-v1: Computer Vision Dataset for Sports Understanding with High Quality Annotations
• Domain Generalization with MixStyle
• Pose-guided Visible Part Matching for Occluded Person ReID
• Learning to Generate Novel Domains for Domain Generalization
• Simple Unsupervised Multi-Object Tracking
• Leveraging Ensembles and Self-Supervised Learning for Fully-Unsupervised Person Re-Identification and Text Authorship Attribution
• MixStyle Neural Networks for Domain Generalization and Adaptation
• Building Computationally Efficient and Well-Generalizing Person Re-Identification Models with Metric Learning
• Person Re-ID through Unsupervised Hypergraph Rank Selection and Fusion
• Deep Domain-Adversarial Image Generation for Domain Generalisation
• Going Deeper into Semi-supervised Person Re-identification
• CoReS: Compatible Representations via Stationarity
• 3D Human Texture Estimation from a Single Image with Transformers
• Symbiotic Adversarial Learning for Attribute-based Person Search
• Visualizing Color-wise Saliency of Black-Box Image Classification Models
• PAMTRI: Pose-Aware Multi-Task Learning for Vehicle Re-Identification Using Highly Randomized Synthetic Data
• Multi-granularity for knowledge distillation