$ account layumi
This repository provides a comprehensive baseline for person/object re-identification (ReID) using PyTorch. It implements a standard pipeline: data preparation (reorganizing images by identity), training with various backbones (ResNet, DenseNet, Swin, HRNet, etc.) and losses (cross-entropy, triplet, circle, contrastive, etc.), feature extraction, and evaluation (Rank-k, mAP). It also includes re-ranking (CPU and GPU-accelerated via custom CUDA kernels) and adversarial training. The code is well-structured and modular, with many self-contained loss functions and augmentation modules that can be reused in other metric learning projects. However, it is research-oriented: paths are hardcoded, dependencies are not fully pinned, and there is no containerized environment. The repository is actively maintained with support for modern architectures and PyTorch features.