End-to-End Trainable Multi-Instance Pose Estimation with Transformers
We propose a new end-to-end trainable approach for multi-instance pose estimation by combining a convolutional neural network with a transformer. We cast multi-instance pose estimation from images as a direct set prediction problem. Inspired by recent work on end-to-end trainable object detection with transformers, we use a transformer encoder-decoder architecture together with a bipartite matching scheme to directly regress the pose of all individuals in a given image. Our model, called POse Estimation Transformer (POET), is trained using a novel set-based global loss that consists of a keypoint loss, a keypoint visibility loss, a center loss and a class loss. POET reasons about the relations between detected humans and the full image context to directly predict the poses in parallel. We show that POET can achieve high accuracy on the challenging COCO keypoint detection task. To the best of our knowledge, this model is the first end-to-end trainable multi-instance human pose estimation method.
Code (2)
Tasks
DecoderKeypoint Detectionobject-detectionObject DetectionPose EstimationMethods 이 논문이 사용한 방법론
Similar Papers 제목 키워드 기반
COPE: End-to-end trainable Constant Runtime Object Pose Estimation
State-of-the-art object pose estimation handles multiple instances in a test image by using multi-model formulations: detection as a first stage and then separately trained networks per object for 2D-3D geometric corresp…
6D Pose Estimation using RGBObjectPose EstimationLearning Spectral Methods by Transformers
Transformers demonstrate significant advantages as the building block of modern LLMs. In this work, we study the capacities of Transformers in performing unsupervised learning. We show that multi-layered Transformers, gi…
In-Context LearningAn End-to-End Trainable Video Panoptic Segmentation Method usingTransformers
In this paper, we present an algorithm to tackle a video panoptic segmentation problem, a newly emerging area of research. The video panoptic segmentation is a task that unifies the typical task of panoptic segmentation …
Multi-Object TrackingObject TrackingPanoptic SegmentationSegmentation+1InsPose: Instance-Aware Networks for Single-Stage Multi-Person Pose Estimation
Multi-person pose estimation is an attractive and challenging task. Existing methods are mostly based on two-stage frameworks, which include top-down and bottom-up methods. Two-stage methods either suffer from high compu…
Multi-Person Pose EstimationPose EstimationCausal Transformer for Fusion and Pose Estimation in Deep Visual Inertial Odometry
In recent years, transformer-based architectures become the de facto standard for sequence modeling in deep learning frameworks. Inspired by the successful examples, we propose a causal visual-inertial fusion transformer…
Pose Estimation