HyperAIHyperAI

Command Palette

Search for a command to run...

CIA-SSD : Détecteur d'objets mono-stade basé sur le nuage de points et sensible à l'IoU avec confiance

Wu Zheng Weiliang Tang Sijin Chen Li Jiang Chi-Wing Fu

Résumé

Les détecteurs mono-stade existants pour la localisation d'objets dans les nuages de points traitent souvent la localisation des objets et leur classification en catégories comme des tâches distinctes, ce qui peut entraîner un décalage entre la précision de la localisation et la confiance de la classification. Pour remédier à ce problème, nous présentons un nouveau détecteur mono-stade nommé le Détecteur Mono-Stade Conscient de l'IoU et Confiant (CIA-SSD). Tout d'abord, nous concevons un module d'agrégation de caractéristiques spatiales et sémantiques léger afin de fusionner de manière adaptative les caractéristiques sémantiques abstraites de haut niveau et les caractéristiques spatiales de bas niveau pour des prédictions précises des boîtes englobantes et de la confiance de classification. De plus, la confiance prédite est corrigée par notre module de correction de confiance conscient de l'IoU, afin que cette confiance soit plus cohérente avec la précision de la localisation. Sur la base de cette confiance corrigée, nous formulons une suppression non maximale (NMS) pondérée par l'IoU variant selon la distance, permettant d'obtenir des régressions plus fluides et d'éviter les prédictions redondantes. Nous expérimentons le CIA-SSD sur la détection 3D des voitures dans l'ensemble de test KITTI et montrons qu'il atteint des performances optimales en termes du métrique officielle du classement (AP modéré 80,28 %) et une vitesse d'inférence supérieure à 32 FPS, surpassant tous les détecteurs mono-stade précédents. Le code est disponible à l'adresse suivante : https://github.com/Vegeta2020/CIA-SSD.


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
CIA-SSD : Détecteur d'objets mono-stade basé sur le nuage de points et sensible à l'IoU avec confiance | Articles | HyperAI