8 天前
FastReID:一个用于通用实例再识别的 PyTorch 工具箱
Lingxiao He, Xingyu Liao, Wu Liu, Xinchen Liu, Peng Cheng, Tao Mei

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