3 个月前

FastReID:一个用于通用实例再识别的 PyTorch 工具箱

FastReID:一个用于通用实例再识别的 PyTorch 工具箱

摘要

通用实例重识别(General Instance Re-identification)是计算机视觉领域中一项至关重要的任务,可广泛应用于诸多实际场景,如行人与车辆重识别、人脸识别、野生动物保护、商品溯源以及快照分析等。为满足日益增长的通用实例重识别应用需求,京东AI研究院推出了FastReID——一个面向广泛使用的软件系统。FastReID采用高度模块化与可扩展的设计架构,使研究人员能够便捷地实现新的研究构想。其友好的系统配置管理与工程化部署功能,也使从业者能够快速将模型投入实际生产环境。目前,我们已在FastReID中实现了多项前沿研究成果,涵盖行人重识别、部分行人重识别、跨域重识别以及车辆重识别等任务,并计划在多个基准数据集上发布预训练模型。截至目前,FastReID是支持单机与多GPU服务器部署的最通用、高性能的工具箱之一,用户可轻松复现我们的实验结果。我们诚挚欢迎广大研究者与开发者使用该系统,相关代码与模型已开源,详见:https://github.com/JDAI-CV/fast-reid。

代码仓库

JDAI-CV/fast-reid
官方
pytorch
linghu8812/tensorrt_tracker
pytorch
GitHub 中提及
Vill-Lab/2023-TIFS-DTIBA
tf
GitHub 中提及

基准测试

基准方法指标
person-re-identification-on-market-1501-cSBS (ResNet-50)
Rank-1: 34.13
mAP: 11.54
mINP: 0.29
person-re-identification-on-msmt17-cSBS (ResNet-50)
Rank-1: 28.77
mAP: 7.89
mINP: 0.05

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
FastReID:一个用于通用实例再识别的 PyTorch 工具箱 | 论文 | HyperAI超神经