HyperAIHyperAI

Command Palette

Search for a command to run...

NMS-Loss : Apprentissage avec la suppression non maximale pour la détection de piétons en situation de forte densité

Zekun Luo Zheng Fang Sixiao Zheng Yabiao Wang Yanwei Fu

Résumé

La suppression non maximale (Non-Maximum Suppression, NMS) est essentielle pour la détection d'objets et influence les résultats d'évaluation en introduisant des faux positifs (FP) et des faux négatifs (FN), particulièrement dans les scènes de foule avec occlusion. Dans cet article, nous mettons en évidence le problème de faible connexion entre les cibles d'apprentissage et les métriques d'évaluation causée par NMS, et proposons une nouvelle perte, appelée NMS-Loss, permettant d'entraîner de manière end-to-end la procédure NMS sans ajouter de paramètres réseau supplémentaires. La NMS-Loss pénalise deux cas : l'échec de suppression d'un faux positif et l'élimination erronée d'un vrai négatif par NMS. Plus précisément, nous introduisons une perte d'attraction (pull loss) pour rapprocher les prédictions associées à la même cible, et une perte de répulsion (push loss) pour éloigner les prédictions associées à des cibles différentes. Les résultats expérimentaux montrent qu’avec l’aide de la NMS-Loss, notre détecteur, nommé NMS-Ped, atteint des performances remarquables, avec un taux de non-détection (Miss Rate) de 5,92 % sur le jeu de données Caltech et de 10,08 % sur CityPersons, surpassant ainsi tous les concurrents de l’état de l’art.


Créer de l'IA avec l'IA

De l'idée au lancement — accélérez votre développement IA avec le co-codage IA gratuit, un environnement prêt à l'emploi et le meilleur prix pour les GPU.

Codage assisté par IA
GPU prêts à l’emploi
Tarifs les plus avantageux

HyperAI Newsletters

Abonnez-vous à nos dernières mises à jour
Nous vous enverrons les dernières mises à jour de la semaine dans votre boîte de réception à neuf heures chaque lundi matin
Propulsé par MailChimp