FastReID: 一个用于通用实例重识别的 PyTorch 工具箱
计算机视觉与模式识别
2020-07-16 v4
摘要
通用实例重识别是计算机视觉中一项非常重要的任务,可广泛应用于许多实际场景,如行人/车辆重识别、人脸识别、野生动物保护、商品追踪和快照等。为满足日益增长的通用实例重识别应用需求,我们推出了在京东 AI 研究院广泛使用的软件系统 FastReID。在 FastReID 中,高度模块化且可扩展的设计使研究人员能够轻松实现新的研究想法。友好的可管理系统配置与工程部署功能让从业者能够快速将模型部署到生产环境中。我们已经实现了一些最先进的项目,包括行人重识别、局部重识别、跨域重识别和车辆重识别,并计划在多个基准数据集上发布这些预训练模型。FastReID 是迄今为止最通用且高性能的工具箱,支持单 GPU 与多 GPU 服务器,你可以非常容易地复现我们的项目结果,非常欢迎使用它,代码与模型可在 https://github.com/JDAI-CV/fast-reid 获取。
引用
@article{arxiv.2006.02631,
title = {FastReID: A Pytorch Toolbox for General Instance Re-identification},
author = {Lingxiao He and Xingyu Liao and Wu Liu and Xinchen Liu and Peng Cheng and Tao Mei},
journal= {arXiv preprint arXiv:2006.02631},
year = {2020}
}