deep-person-reid

收录理由

行人重识别要解决的是在不同摄像头画面中辨认同一个人的问题，而Torchreid把这一领域里相当多的模型和训练方案集中到了一个PyTorch代码库中。你不需要把零散的研究代码拼凑起来，就能直接训练模型、做评估，并和其他方法对比。它同时支持基于图像和视频的重识别，也照顾到了多GPU训练和数据集准备这些环节，跑通一个基线模型会快很多。代码里还带了跨数据集评估和重排序工具，方便你检验模型在训练集之外的表现。对于想要一个研究级起点、而不是黑盒API的团队来说，这套代码结构清晰，文档也写得比较完整。