FAST GDRNPP: Improving the Speed of State-of-the-Art 6D Object Pose Estimation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Pöllabauer, Thomas, Pramod, Ashwin, Knauthe, Volker, Wahl, Michael
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913508290985984
author Pöllabauer, Thomas
Pramod, Ashwin
Knauthe, Volker
Wahl, Michael
author_facet Pöllabauer, Thomas
Pramod, Ashwin
Knauthe, Volker
Wahl, Michael
contents 6D object pose estimation involves determining the three-dimensional translation and rotation of an object within a scene and relative to a chosen coordinate system. This problem is of particular interest for many practical applications in industrial tasks such as quality control, bin picking, and robotic manipulation, where both speed and accuracy are critical for real-world deployment. Current models, both classical and deep-learning-based, often struggle with the trade-off between accuracy and latency. Our research focuses on enhancing the speed of a prominent state-of-the-art deep learning model, GDRNPP, while keeping its high accuracy. We employ several techniques to reduce the model size and improve inference time. These techniques include using smaller and quicker backbones, pruning unnecessary parameters, and distillation to transfer knowledge from a large, high-performing model to a smaller, more efficient student model. Our findings demonstrate that the proposed configuration maintains accuracy comparable to the state-of-the-art while significantly improving inference time. This advancement could lead to more efficient and practical applications in various industrial scenarios, thereby enhancing the overall applicability of 6D Object Pose Estimation models in real-world settings.
format Preprint
id arxiv_https___arxiv_org_abs_2409_12720
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle FAST GDRNPP: Improving the Speed of State-of-the-Art 6D Object Pose Estimation
Pöllabauer, Thomas
Pramod, Ashwin
Knauthe, Volker
Wahl, Michael
Computer Vision and Pattern Recognition
Artificial Intelligence
6D object pose estimation involves determining the three-dimensional translation and rotation of an object within a scene and relative to a chosen coordinate system. This problem is of particular interest for many practical applications in industrial tasks such as quality control, bin picking, and robotic manipulation, where both speed and accuracy are critical for real-world deployment. Current models, both classical and deep-learning-based, often struggle with the trade-off between accuracy and latency. Our research focuses on enhancing the speed of a prominent state-of-the-art deep learning model, GDRNPP, while keeping its high accuracy. We employ several techniques to reduce the model size and improve inference time. These techniques include using smaller and quicker backbones, pruning unnecessary parameters, and distillation to transfer knowledge from a large, high-performing model to a smaller, more efficient student model. Our findings demonstrate that the proposed configuration maintains accuracy comparable to the state-of-the-art while significantly improving inference time. This advancement could lead to more efficient and practical applications in various industrial scenarios, thereby enhancing the overall applicability of 6D Object Pose Estimation models in real-world settings.
title FAST GDRNPP: Improving the Speed of State-of-the-Art 6D Object Pose Estimation
topic Computer Vision and Pattern Recognition
Artificial Intelligence
url https://arxiv.org/abs/2409.12720