設計では特定のドメイン知識は考慮されていませんが、プレーンビジョントランスフォーマーは視覚認識タスクで優れたパフォーマンスを示しています。ただし、ポーズ推定タスクのためのそのような単純な構造の可能性を明らかにするための努力はほとんど行われていません。この論文では、モデル構造の単純さ、モデルサイズのスケーラビリティ、トレーニングパラダイムの柔軟性、モデル間の知識の伝達可能性など、さまざまな側面からポーズを推定するためのプレーンビジョントランスの驚くほど優れた機能を、と呼ばれる単純なベースラインモデルを通じて示します。 ViTPose。具体的には、ViTPoseは、特定の人物インスタンスの特徴を抽出するためのバックボーンとしてプレーンおよび非階層型のビジョントランスフォーマーを採用し、ポーズ推定のための軽量デコーダーを採用しています。スケーラブルなモデル容量と変圧器の高い並列処理の利点を活用し、スループットとパフォーマンスの間に新しいパレートフロントを設定することで、100Mから1Bのパラメーターにスケールアップできます。さらに、ViTPoseは、注意の種類、入力の解像度、事前トレーニング、微調整戦略、および複数のポーズタスクの処理に関して非常に柔軟性があります。また、大きなViTPoseモデルの知識は、単純な知識トークンを介して小さなモデルに簡単に転送できることを経験的に示しています。実験結果は、私たちの基本的なViTPoseモデルが、挑戦的なMS COCOキーポイント検出ベンチマークの代表的な方法を上回っていることを示していますが、最大のモデルは新しい最先端を設定しています。コードとモデルはhttps://github.com/ViTAE-Transformer/ViTPoseで入手できます。
Although no specific domain knowledge is considered in the design, plain vision transformers have shown excellent performance in visual recognition tasks. However, little effort has been made to reveal the potential of such simple structures for pose estimation tasks. In this paper, we show the surprisingly good capabilities of plain vision transformers for pose estimation from various aspects, namely simplicity in model structure, scalability in model size, flexibility in training paradigm, and transferability of knowledge between models, through a simple baseline model called ViTPose. Specifically, ViTPose employs plain and non-hierarchical vision transformers as backbones to extract features for a given person instance and a lightweight decoder for pose estimation. It can be scaled up from 100M to 1B parameters by taking the advantages of the scalable model capacity and high parallelism of transformers, setting a new Pareto front between throughput and performance. Besides, ViTPose is very flexible regarding the attention type, input resolution, pre-training and finetuning strategy, as well as dealing with multiple pose tasks. We also empirically demonstrate that the knowledge of large ViTPose models can be easily transferred to small ones via a simple knowledge token. Experimental results show that our basic ViTPose model outperforms representative methods on the challenging MS COCO Keypoint Detection benchmark, while the largest model sets a new state-of-the-art. The code and models are available at https://github.com/ViTAE-Transformer/ViTPose.