HyperAIHyperAI

Command Palette

Search for a command to run...

Segmentation sémantique 3D efficace avec Superpoint Transformer

Damien Robert Hugo Raguet Loic Landrieu

Résumé

Nous introduisons une nouvelle architecture transformer basée sur des superpoints pour une segmentation sémantique efficace des scènes 3D à grande échelle. Notre méthode intègre un algorithme rapide permettant de partitionner les nuages de points en une structure hiérarchique de superpoints, rendant notre prétraitement jusqu’à 7 fois plus rapide que les approches existantes basées sur les superpoints. En outre, nous exploitons un mécanisme d’attention auto-associative pour capturer les relations entre les superpoints à plusieurs échelles, ce qui conduit à des performances de pointe sur trois jeux de données de référence exigeants : S3DIS (76,0 % de mIoU sur validation 6-fold), KITTI-360 (63,5 % sur Val) et DALES (79,6 %). Avec seulement 212 000 paramètres, notre approche est jusqu’à 200 fois plus compacte que d’autres modèles de pointe tout en maintenant des performances comparables. En outre, notre modèle peut être entraîné sur une seule GPU en 3 heures pour une fold du jeu de données S3DIS, soit entre 7 et 70 fois moins d’heures de GPU par rapport aux méthodes les plus performantes. Notre code et nos modèles sont disponibles à l’adresse github.com/drprojects/superpoint_transformer.


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