From Points to Multi-Object 3D Reconstruction
This work addresses the problem of real-time, multi-object 3D reconstruction from a single image, which is important for applications in robotics and augmented reality, offering an incremental improvement in speed and realism.
This paper introduces a method for detecting and reconstructing multiple 3D objects from a single RGB image. It jointly optimizes for detection, alignment, and shape, predicting 9-DoF bounding boxes and 3D shapes in a single forward pass by selecting from exemplar shapes and using a collision-loss for realism.
We propose a method to detect and reconstruct multiple 3D objects from a single RGB image. The key idea is to optimize for detection, alignment and shape jointly over all objects in the RGB image, while focusing on realistic and physically plausible reconstructions. To this end, we propose a keypoint detector that localizes objects as center points and directly predicts all object properties, including 9-DoF bounding boxes and 3D shapes -- all in a single forward pass. The proposed method formulates 3D shape reconstruction as a shape selection problem, i.e. it selects among exemplar shapes from a given database. This makes it agnostic to shape representations, which enables a lightweight reconstruction of realistic and visually-pleasing shapes based on CAD-models, while the training objective is formulated around point clouds and voxel representations. A collision-loss promotes non-intersecting objects, further increasing the reconstruction realism. Given the RGB image, the presented approach performs lightweight reconstruction in a single-stage, it is real-time capable, fully differentiable and end-to-end trainable. Our experiments compare multiple approaches for 9-DoF bounding box estimation, evaluate the novel shape-selection mechanism and compare to recent methods in terms of 3D bounding box estimation and 3D shape reconstruction quality.