HyperAIHyperAI

Command Palette

Search for a command to run...

Réseaux Squeeze-and-Attention pour la segmentation sémantique

Zilong Zhong Zhong Qiu Lin Rene Bidart Xiaodan Hu Ibrahim Ben Daya Zhifeng Li Wei-Shi Zheng Jonathan Li Alexander Wong

Résumé

L’intégration récente de mécanismes d’attention dans les réseaux de segmentation améliore leurs capacités de représentation en mettant l’accent sur des caractéristiques plus informatives. Toutefois, ces mécanismes d’attention négligent une sous-tâche implicite de la segmentation sémantique et sont limités par la structure en grille des noyaux de convolution. Dans cet article, nous proposons une nouvelle architecture de réseau de type squeeze-and-attention (SANet), qui exploite un module efficace de squeeze-and-attention (SA) afin de tenir compte de deux caractéristiques distinctives de la segmentation : i) l’attention sur des groupes de pixels, et ii) la prédiction au niveau pixel. Plus précisément, les modules SA proposés imposent une attention sur les groupes de pixels aux convolutions conventionnelles en introduisant un canal de convolution « d’attention », permettant ainsi de modéliser efficacement les dépendances spatiales et canaliques. Les résultats finaux de segmentation sont obtenus en fusionnant les sorties provenant des quatre étages hiérarchiques de SANet, afin d’intégrer des contextes multi-échelles et d’améliorer ainsi la prédiction au niveau pixel. Des expérimentations empiriques menées sur deux jeux de données publics exigeants valident l’efficacité des SANets proposés, qui atteignent un mIoU de 83,2 % (sans pré-entraînement sur COCO) sur PASCAL VOC et un mIoU de pointe de 54,4 % sur PASCAL Context.


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