Search papers, labs, and topics across Lattice.
Yolo-Key-6D, a novel single-stage framework, is introduced for monocular 6D pose estimation, targeting real-time applications. The architecture enhances a YOLO-based detector with an auxiliary head that regresses 2D projections of 3D bounding box corners to improve geometric understanding. Rotation is directly regressed using a continuous 9D representation projected to SO(3) via singular value decomposition, achieving competitive accuracy on LINEMOD and LINEMOD-Occluded datasets while maintaining real-time performance.
Ditch the latency: Yolo-Key-6D achieves competitive 6D pose estimation accuracy in real-time using a single-stage architecture.
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.