HyperAIHyperAI

Command Palette

Search for a command to run...

SOLQ : Segmenter des objets en apprenant des requêtes

Bin Dong Fangao Zeng Tiancai Wang Xiangyu Zhang Yichen Wei

Résumé

Dans cet article, nous proposons un cadre end-to-end pour la segmentation d'instances. Inspiré du modèle récemment introduit DETR [1], notre méthode, nommée SOLQ, segmente les objets en apprenant des requêtes unifiées. Dans SOLQ, chaque requête représente un objet et possède plusieurs représentations : classe, localisation et masque. Les requêtes d'objets apprises effectuent simultanément la classification, la régression de boîtes englobantes et le codage de masques sous une forme vectorielle unifiée. Pendant la phase d'entraînement, les vecteurs de masques codés sont supervisés par le codage par compression des masques spatiaux bruts. Lors de l'inférence, les vecteurs de masques générés peuvent être directement transformés en masques spatiaux grâce au processus inverse du codage par compression. Les résultats expérimentaux montrent que SOLQ atteint des performances de pointe, dépassant la plupart des approches existantes. De plus, l'apprentissage conjoint de la représentation unifiée des requêtes améliore considérablement les performances de détection de DETR. Nous espérons que SOLQ pourra servir de base solide pour la segmentation d'instances basée sur les Transformers. Le code est disponible à l'adresse suivante : https://github.com/megvii-research/SOLQ.


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
SOLQ : Segmenter des objets en apprenant des requêtes | Articles | HyperAI