行人再识别Person-reID的Pytorch实现
Person_reID_baseline_pytorch 一个小巧、友好、强大的 Person-reID 基线代码(基于 pytorch)。
强的。
它与几个顶级会议工作中的新基线结果一致,例如,用于人员重新识别的联合判别和生成学习(CVPR19)、超越部件模型:具有精细部件池的人员检索(ECCV18)、用于人员的相机风格适应重新识别(CVPR18)。
我们到达 Rank@1=88.24%, mAP=70.68% 仅使用 softmax 损失。
小的。
使用 fp16(由 Nvidia apex 支持),我们的基线可以仅使用 2GB GPU 内存进行训练。
友谊赛。
您可以使用现成的选项在一行中应用许多最先进的技巧。
此外,如果您不熟悉 person re-ID,您可以先查看我们的教程(8 分钟阅读):thumbs_up:。
目录 特性 一些新闻 训练模型先决条件 入门 安装 数据集准备 训练测试评估 使用其他数据集进行训练的技巧 引文 相关存储库功能 现在我们支持: Float16 以节省 GPU 内存,基于 a
2021-08-26 17:32:38
271KB
机器学习
1