cs.CVMar 4, 2026

Yolo-Key-6D: Single Stage Monocular 6D Pose Estimation with Keypoint Enhancements

Authors: Kemal Alperen Çetiner, Hazım Kemal Ekenel

Organizations: ASELSAN, Aerospace Technologies, Türkiye · Istanbul Technical University, Department of Computer Engineering, Türkiye · New York University Abu Dhabi, Division of Engineering, UAE

Abstract

Estimating the 6D pose of objects from a single RGB image is a critical task for robotics and extended reality applications. However, state-of-the-art multi stage methods often suffer from high latency, making them unsuitable for real time use. In this paper, we present Yolo-Key-6D, a novel single stage, end-to-end framework for monocular 6D pose estimation designed for both speed and accuracy. Our approach enhances a YOLO based architecture by integrating an auxiliary head that regresses the 2D projections of an object's 3D bounding box corners. This keypoint detection task significantly improves the network's understanding of 3D geometry. For stable end-to-end training, we directly regress rotation using a continuous 9D representation projected to SO(3) via singular value decomposition. On the LINEMOD and LINEMOD-Occluded benchmarks, YOLO-Key-6D achieves competitive accuracy scores of 96.24% and 69.41%, respectively, with the ADD(-S) 0.1d metric, while proving itself to operate in real time. Our results demonstrate that a carefully designed single stage method can provide a practical and effective balance of performance and efficiency for real world deployment.

Figures & tables

Explore similar work

CardsList
  1. TinyDETR-Pose: Towards End-to-End Real-Time Single-Stage 6DoF Object Pose Estimation with Lightweight Transformers

    Aug 15, 2026Paul Julius Kühn, Duc Anh Nguyen, Saptarshi Neil Sinha +2Object Pose EstimationEdge Inference

  2. 6D Pose Estimation via Keypoint Heatmap Regression with RGB-D Residual Neural Networks

    May 8, 2026Ismail Aljosevic, Amir Masoud Almasi, Ana Parovic +1Object Pose Estimation