HyperAIHyperAI

Command Palette

Search for a command to run...

RTMO : Vers une estimation de pose multi-personne en temps réel à haute performance en une seule étape

Peng Lu Tao Jiang Yining Li Xiangtai Li Kai Chen Wenming Yang

Résumé

L'estimation en temps réel de la posture de plusieurs personnes présente des défis importants en matière d'équilibre entre vitesse et précision. Bien que les méthodes top-down en deux étapes ralentissent avec l'augmentation du nombre de personnes dans l'image, les méthodes existantes en une étape échouent souvent à fournir simultanément une haute précision et des performances en temps réel. Cet article introduit RTMO, un cadre d'estimation de la posture en une étape qui intègre sans heurt la classification des coordonnées en représentant les points clés par des cartes thermiques 1-D duals au sein de l'architecture YOLO, atteignant une précision comparable aux méthodes top-down tout en maintenant une vitesse élevée. Nous proposons un classificateur de coordonnées dynamique et une fonction de perte spécifique pour l'apprentissage des cartes thermiques, conçus spécifiquement pour résoudre les incompatibilités entre la classification des coordonnées et les modèles de prédiction dense. RTMO surpasse les estimateurs de posture en une étape les plus avancés actuellement disponibles, obtenant un AP 1,1% plus élevé sur COCO tout en opérant environ 9 fois plus rapidement avec le même backbone. Notre modèle le plus performant, RTMO-l, atteint 74,8% d'AP sur COCO val2017 et 141 FPS sur une seule GPU V100, démontrant ainsi son efficacité et sa précision. Le code source et les modèles sont disponibles à l'adresse suivante : https://github.com/open-mmlab/mmpose/tree/main/projects/rtmo.


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