HyperAIHyperAI
il y a 8 jours

FastReID : Une boîte à outils PyTorch pour la réidentification d'instances généralisées

Lingxiao He, Xingyu Liao, Wu Liu, Xinchen Liu, Peng Cheng, Tao Mei
FastReID : Une boîte à outils PyTorch pour la réidentification d'instances généralisées
Résumé

La réidentification d'instances générales est une tâche fondamentale en vision par ordinateur, pouvant être largement appliquée à diverses applications pratiques, telles que la réidentification de personnes ou de véhicules, la reconnaissance faciale, la protection de la faune sauvage, le traçage des produits ou encore la capture instantanée (snapshot). Afin de répondre à la demande croissante en matière de réidentification d'instances générales, nous présentons FastReID, un système logiciel largement utilisé au sein de JD AI Research. Grâce à une conception hautement modulaire et extensible, FastReID permet aux chercheurs d'implémenter facilement de nouvelles idées scientifiques. Ses fonctionnalités d'administration conviviales et de déploiement ingénieristique permettent aux praticiens de déployer rapidement des modèles en production. Nous avons implémenté plusieurs projets de pointe, notamment la réidentification de personnes, la réidentification partielle, la réidentification trans-domaine et la réidentification de véhicules, et prévoyons de publier prochainement ces modèles pré-entraînés sur plusieurs jeux de données de référence. FastReID constitue actuellement la boîte à outils la plus générale et performante, prenant en charge à la fois les serveurs à GPU unique et multiples. Il est désormais très facile de reproduire nos résultats, et nous vous invitons chaleureusement à l'utiliser : le code source et les modèles sont disponibles à l'adresse suivante : https://github.com/JDAI-CV/fast-reid.