ReFlow6D: Refraction-Guided Transparent Object 6D Pose Estimation via Intermediate Representation Learning
Hrishikesh Gupta, Stefan Thalhammer, Jean-Baptiste Weibel, Alexander Haberl, Markus Vincze
TL;DR
This work tackles the challenging problem of monocular 6D pose estimation for transparent objects, whose appearance is strongly affected by light refraction. It introduces ReFlow6D, which learns a refractive-intermediate representation comprising refractive flow, attenuation, surface-region attention, and a visibility mask to encode environment-invariant light deformation. A Patch-PnP-based regressor uses these features for direct 6D pose estimation, augmented by a novel object compositing loss to refine the intermediate representations. Empirical results on TOD and Trans6D-32K show state-of-the-art accuracy, and real-world grasping experiments demonstrate practical applicability, underscoring the advantage of refractive-based features over traditional geometric or edge cues.
Abstract
Transparent objects are ubiquitous in daily life, making their perception and robotics manipulation important. However, they present a major challenge due to their distinct refractive and reflective properties when it comes to accurately estimating the 6D pose. To solve this, we present ReFlow6D, a novel method for transparent object 6D pose estimation that harnesses the refractive-intermediate representation. Unlike conventional approaches, our method leverages a feature space impervious to changes in RGB image space and independent of depth information. Drawing inspiration from image matting, we model the deformation of the light path through transparent objects, yielding a unique object-specific intermediate representation guided by light refraction that is independent of the environment in which objects are observed. By integrating these intermediate features into the pose estimation network, we show that ReFlow6D achieves precise 6D pose estimation of transparent objects, using only RGB images as input. Our method further introduces a novel transparent object compositing loss, fostering the generation of superior refractive-intermediate features. Empirical evaluations show that our approach significantly outperforms state-of-the-art methods on TOD and Trans32K-6D datasets. Robot grasping experiments further demonstrate that ReFlow6D's pose estimation accuracy effectively translates to real-world robotics task. The source code is available at: https://github.com/StoicGilgamesh/ReFlow6D and https://github.com/StoicGilgamesh/matting_rendering.
