8日前

FastReID:一般インスタンス再識別向けPyTorchツールボックス

Lingxiao He, Xingyu Liao, Wu Liu, Xinchen Liu, Peng Cheng, Tao Mei
FastReID:一般インスタンス再識別向けPyTorchツールボックス
要約

一般インスタンス再識別(General Instance Re-identification)は、コンピュータビジョン分野において極めて重要なタスクであり、人物・車両の再識別、顔認識、野生生物保護、商品トレーサビリティ、スナップショット解析など、多くの実用的応用に活用されています。こうした一般インスタンス再識別に対する応用ニーズの高まりに応えるため、JD AI Researchにて広く利用可能なソフトウェアシステム「FastReID」を提案します。FastReIDは、高いモジュール性と拡張性を備えており、研究者が新しい研究アイデアを迅速に実装できるよう設計されています。また、直感的なシステム設定機能とエンジニアリング的なデプロイ支援機能により、実務家がモデルを生産環境に素早く展開することが可能になります。現在までに、人物再識別(person re-id)、部分領域再識別(partial re-id)、クロスドメイン再識別(cross-domain re-id)、車両再識別(vehicle re-id)など、最先端のプロジェクトを実装しており、複数のベンチマークデータセット上で事前学習済みモデルの公開を予定しています。FastReIDは、単一GPUおよび複数GPUサーバーをサポートする、これまでで最も汎用性が高く高性能なツールボックスです。本研究の成果を容易に再現でき、ぜひご活用ください。コードおよびモデルは、以下のGitHubリポジトリにて公開されています:https://github.com/JDAI-CV/fast-reid。