Neural network techniques for predicting object poses based on parameterized and non-parameterized rotation and translation vectors

Abstract

Disclosed herein are systems and methods that provide an end-to-end approach for performing multi-dimensional object pose estimation in the context of machine learning models. In an implementation, processing circuitry of a suitable computer inputs image data to a machine learning model that predicts a parameterized rotation vector and a parameterized translation vector for an object in the image. Next, the processing circuitry converts the parameterized rotation vector and the parameterized translation vector into a non-parameterized rotation vector and a non-parameterized translation vector respectively. Finally, the processing circuitry updates the image data based on the non-parameterized rotation vector and the non-parameterized translation vector.