HyperAIHyperAI

Command Palette

Search for a command to run...

Régression de boîtes englobantes avec incertitude pour une détection d'objets précise

Yihui He Chenchen Zhu Jianren Wang Marios Savvides Xiangyu Zhang

Résumé

Les jeux de données d'annotation d'objets à grande échelle (par exemple, MS-COCO) tentent de définir les boîtes englobantes de vérité terrain aussi clairement que possible. Cependant, nous constatons que des ambiguïtés subsistent lors de l'étiquetage des boîtes englobantes. Dans cet article, nous proposons une nouvelle perte de régression pour les boîtes englobantes permettant d'apprendre la transformation et la variance de localisation ensemble. Notre perte améliore considérablement la précision de localisation de diverses architectures avec presque aucune augmentation du calcul. La variance de localisation apprise nous permet de fusionner les boîtes englobantes voisines lors de la suppression non maximale (NMS), ce qui améliore encore davantage les performances de localisation. Sur MS-COCO, nous augmentons la Précision Moyenne (AP) du VGG-16 Faster R-CNN de 23,6 % à 29,1 %. Plus important encore, pour le ResNet-50-FPN Mask R-CNN, notre méthode améliore respectivement l'AP et l'AP90 de 1,8 % et 6,2 %, surpassant significativement les méthodes précédentes d'affinement des boîtes englobantes. Notre code et nos modèles sont disponibles sur : github.com/yihui-he/KL-Loss


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