3D Pose Estimation With Simulated Training and Occlusion Handling

Resolve Bottlenecks,
Find Innovative Solutions
Generate Solutions

Solution Overview

Problem

Acquiring and labeling real-world data for 3D pose estimation in robotics is challenging, time-consuming, and prone to error, especially for specialized scenarios, leading to slow adoption of models and inaccurate results due to computational overhead and occlusions.

Innovation Solution

Training autoencoders using simulation data to predict 3D pose labels, which includes an object detector and an autoencoder that predicts segmentation images invariant to occlusions, and adapting models with Generative Adversarial Networks to differentiate between simulated and real-world data.

Engineering Contradictions & Design Principles

VSEngineering Contradiction Analysis

1Quantity of substance

If conventional systems use a trained augmented autoencoder with codebook generation and lookup for 3D pose estimation, then labeled training data requirements are reduced, but computational overhead increases and accuracy decreases in specialized scenarios

Engineering Contradiction:
Improvelabeled training dataVSAvoidcomputational overhead
Core Design Contradiction:
Quantity of substanceVSDevice complexity

Solution Approach 1:

The patent extracts and removes the codebook generation and lookup mechanism from the autoencoder system. Instead of using a separate codebook for decoding, the system directly processes images through the autoencoder to obtain 3D pose estimates, eliminating the computational overhead associated with codebook maintenance and comparison while preserving the core functionality of unsupervised learning from unlabeled data

Inventive Principle:
Principle #2Taking out (Extraction)

Solution Approach 2:

The patent enhances the autoencoder to handle multiple functions within a single unified architecture. The model simultaneously performs 3D pose estimation, occlusion handling, and adaptation to specialized scenarios without requiring separate codebooks or multiple specialized models, thereby reducing computational overhead while maintaining versatility across different applications

Inventive Principle:
Principle #6Universality (Multi-functionality)

2Ease of operation

If conventional systems use bounding box comparison to determine object center for 3D pose, then the process is simple, but accuracy decreases when occlusions are present

Engineering Contradiction:
Improveobject center determinationVSAvoidobject center accuracy
Core Design Contradiction:
Ease of operationVSMeasurement precision

Solution Approach 1:

The patent applies preliminary actions by training the autoencoder to predict segmentation masks and handle occlusions before determining the object center. The model first processes the input image through the autoencoder to obtain a refined representation that accounts for occlusions, then uses this enhanced representation to accurately determine the object center, rather than directly comparing bounding boxes which would be inaccurate under occlusion

Inventive Principle:
Principle #10Preliminary action

Solution Approach 2:

The patent introduces an intermediary mechanism in the form of the autoencoder's latent space and segmentation predictions. Instead of directly comparing bounding boxes, the system uses the autoencoder's learned representations and segmentation masks as intermediaries to infer the object center, thereby achieving accurate center determination even when occlusions are present

Inventive Principle:
Principle #24Intermediary (Mediator)

3Reliability

If augmented autoencoders are trained with various noise augmentations to enhance robustness, then noise robustness improves, but the number of parameters required increases significantly

Engineering Contradiction:
Improvenoise robustnessVSAvoidnumber of parameters
Core Design Contradiction:
ReliabilityVSDevice complexity

Solution Approach 1:

The patent applies partial action by selectively applying noise robustness techniques only where needed. Instead of training with all possible noise augmentations across the entire system, the approach focuses on training the autoencoder with representative noise patterns and occlusions, achieving sufficient robustness without the excessive parameter inflation that would result from comprehensive noise augmentation training

Inventive Principle:
Principle #16Partial or excessive action

Data Source

PatentUS11823415B23D pose estimation in robotics
Publication Date: 2023.11.21 NVIDIA CORP
  • US11823415B2 patent drawing
  • US11823415B2 patent drawing
  • US11823415B2 patent drawing

AI summary

An autoencoder may be trained to predict 3D pose labels using simulation data extracted from a simulated environment, which may be configured to represent an environment in which the 3D pose estimator is to be deployed. Assets may be used to mimic the deployment environment such as 3D models or textures and parameters used to define deployment scenarios and/or conditions that the 3D pose estimator will operate under in the environment. The autoencoder may be trained to predict a segmentation image from an input image that is invariant to occlusions. Further, the autoencoder may be trained to exclude areas of the input image from the object that correspond to one or more appendages of the object. The 3D pose may be adapted to unlabeled real-world data using a GAN, which predicts whether output of the 3D pose estimator was generated from real-world data or simulated data.