HyperAIHyperAI

Command Palette

Search for a command to run...

Réinterroger la détection d'objets en rotation à l'aide d'une perte basée sur la distance de Wasserstein gaussienne

Xue Yang Junchi Yan Qi Ming Wentao Wang Xiaopeng Zhang Qi Tian

Résumé

La discontinuité de frontière et son incohérence par rapport au métrique de détection finale constituent un goulot d’étranglement majeur dans la conception des pertes de régression pour la détection orientée. Dans cet article, nous proposons une nouvelle perte de régression fondée sur la distance de Wasserstein gaussienne, une approche fondamentale visant à résoudre ce problème. Plus précisément, la boîte englobante orientée est convertie en une distribution gaussienne bidimensionnelle, permettant ainsi d’approcher efficacement la perte induite par l’IoU rotationnel, qui est non différentiable, grâce à la distance de Wasserstein gaussienne (GWD), une fonction pouvant être apprise de manière efficace par propagation arrière du gradient. La GWD reste informative pour l’apprentissage même lorsque deux boîtes englobantes orientées ne s’intersectent pas, situation fréquente dans la détection d’objets de petite taille. Grâce à ses trois propriétés uniques, la GWD permet également de résoudre élégamment les problèmes de discontinuité de frontière et de forme carrée, indépendamment de la manière dont la boîte englobante est définie. Des expériences menées sur cinq jeux de données utilisant différents détecteurs démontrent l’efficacité de notre approche. Le code est disponible à l’adresse suivante : https://github.com/yangxue0827/RotationDetection et https://github.com/open-mmlab/mmrotate.


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
Réinterroger la détection d'objets en rotation à l'aide d'une perte basée sur la distance de Wasserstein gaussienne | Articles | HyperAI