HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

3D Sketch-aware Semantic Scene Completion via Semi-supervised Structure Prior

Chen Xiaokang ; Lin Kwan-Yee ; Qian Chen ; Zeng Gang ; Li Hongsheng

3D Sketch-aware Semantic Scene Completion via Semi-supervised Structure
  Prior

Abstract

The goal of the Semantic Scene Completion (SSC) task is to simultaneouslypredict a completed 3D voxel representation of volumetric occupancy andsemantic labels of objects in the scene from a single-view observation. Sincethe computational cost generally increases explosively along with the growth ofvoxel resolution, most current state-of-the-arts have to tailor their frameworkinto a low-resolution representation with the sacrifice of detail prediction.Thus, voxel resolution becomes one of the crucial difficulties that lead to theperformance bottleneck. In this paper, we propose to devise a new geometry-based strategy to embeddepth information with low-resolution voxel representation, which could stillbe able to encode sufficient geometric information, e.g., room layout, object'ssizes and shapes, to infer the invisible areas of the scene with wellstructure-preserving details. To this end, we first propose a novel 3Dsketch-aware feature embedding to explicitly encode geometric informationeffectively and efficiently. With the 3D sketch in hand, we further devise asimple yet effective semantic scene completion framework that incorporates alight-weight 3D Sketch Hallucination module to guide the inference of occupancyand the semantic labels via a semi-supervised structure prior learningstrategy. We demonstrate that our proposed geometric embedding works betterthan the depth feature learning from habitual SSC frameworks. Our final modelsurpasses state-of-the-arts consistently on three public benchmarks, which onlyrequires 3D volumes of 60 x 36 x 60 resolution for both input and output. Thecode and the supplementary material will be available athttps://charlesCXK.github.io.

Code Repositories

charlesCXK/TorchSSC
pytorch
Mentioned in GitHub
charlesCXK/3D-SketchAware-SSC
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
3d-semantic-scene-completion-from-a-single3DSketch (rgb input - reported in MonoScene paper)
mIoU: 22.91
3d-semantic-scene-completion-from-a-single-13DSketch (rgb input - reported in MonoScene paper)
mIoU: 6.23
3d-semantic-scene-completion-on-nyuv23DSketch
mIoU: 41.1

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp