RGB 画像からの剛体オブジェクトの正確な 6D 姿勢推定は、ロボット工学、拡張現実、および人間とコンピューターのインタラクションにおいて重要ですが困難なタスクです。この問題に対処するために、私たちは 6D ポーズ洗練のための新しいリカレント ネットワーク アーキテクチャである DeepRM を提案します。 DeepRM は、初期の粗い姿勢推定を活用して、ターゲット オブジェクトの合成画像をレンダリングします。次に、レンダリングされたイメージが観察されたイメージと照合され、以前の姿勢推定を更新するための剛体変換が予測されます。このプロセスを繰り返して、反復ごとに推定値を段階的に改良します。 DeepRM アーキテクチャには LSTM ユニットが組み込まれており、各改良ステップを通じて情報を伝達し、全体的なパフォーマンスを大幅に向上させます。現在の 2 段階の Perspective-n-Point ベースのソリューションとは対照的に、DeepRM はエンドツーエンドでトレーニングされ、単一のパラメーターで精度と効率を調整できるスケーラブルなバックボーンを使用します。トレーニング中に、観察画像と合成画像の間のオプティカル フローを予測するために、マルチスケール オプティカル フロー ヘッドが追加されます。オプティカル フロー予測はトレーニング プロセスを安定させ、姿勢推定のタスクに関連する特徴の学習を強制します。私たちの結果は、DeepRM が広く受け入れられている 2 つの困難なデータセットで最先端のパフォーマンスを達成していることを示しています。
Precise 6D pose estimation of rigid objects from RGB images is a critical but challenging task in robotics, augmented reality and human-computer interaction. To address this problem, we propose DeepRM, a novel recurrent network architecture for 6D pose refinement. DeepRM leverages initial coarse pose estimates to render synthetic images of target objects. The rendered images are then matched with the observed images to predict a rigid transform for updating the previous pose estimate. This process is repeated to incrementally refine the estimate at each iteration. The DeepRM architecture incorporates LSTM units to propagate information through each refinement step, significantly improving overall performance. In contrast to current 2-stage Perspective-n-Point based solutions, DeepRM is trained end-to-end, and uses a scalable backbone that can be tuned via a single parameter for accuracy and efficiency. During training, a multi-scale optical flow head is added to predict the optical flow between the observed and synthetic images. Optical flow prediction stabilizes the training process, and enforces the learning of features that are relevant to the task of pose estimation. Our results demonstrate that DeepRM achieves state-of-the-art performance on two widely accepted challenging datasets.