English

HRPose: Real-Time High-Resolution 6D Pose Estimation Network Using Knowledge Distillation

Computer Vision and Pattern Recognition 2022-04-21 v1 Machine Learning Robotics

Abstract

Real-time 6D object pose estimation is essential for many real-world applications, such as robotic grasping and augmented reality. To achieve an accurate object pose estimation from RGB images in real-time, we propose an effective and lightweight model, namely High-Resolution 6D Pose Estimation Network (HRPose). We adopt the efficient and small HRNetV2-W18 as a feature extractor to reduce computational burdens while generating accurate 6D poses. With only 33\% of the model size and lower computational costs, our HRPose achieves comparable performance compared with state-of-the-art models. Moreover, by transferring knowledge from a large model to our proposed HRPose through output and feature-similarity distillations, the performance of our HRPose is improved in effectiveness and efficiency. Numerical experiments on the widely-used benchmark LINEMOD demonstrate the superiority of our proposed HRPose against state-of-the-art methods.

Keywords

Cite

@article{arxiv.2204.09429,
  title  = {HRPose: Real-Time High-Resolution 6D Pose Estimation Network Using Knowledge Distillation},
  author = {Qi Guan and Zihao Sheng and Shibei Xue},
  journal= {arXiv preprint arXiv:2204.09429},
  year   = {2022}
}

Comments

8 pages, 4 figures, and 5 tables, accepted by Chinese Journal of Electronics

R2 v1 2026-06-24T10:53:16.533Z