HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

Soft Rasterizer: A Differentiable Renderer for Image-based 3D Reasoning

Shichen Liu; Tianye Li; Weikai Chen; Hao Li

Soft Rasterizer: A Differentiable Renderer for Image-based 3D Reasoning

Abstract

Rendering bridges the gap between 2D vision and 3D scenes by simulating the physical process of image formation. By inverting such renderer, one can think of a learning approach to infer 3D information from 2D images. However, standard graphics renderers involve a fundamental discretization step called rasterization, which prevents the rendering process to be differentiable, hence able to be learned. Unlike the state-of-the-art differentiable renderers, which only approximate the rendering gradient in the back propagation, we propose a truly differentiable rendering framework that is able to (1) directly render colorized mesh using differentiable functions and (2) back-propagate efficient supervision signals to mesh vertices and their attributes from various forms of image representations, including silhouette, shading and color images. The key to our framework is a novel formulation that views rendering as an aggregation function that fuses the probabilistic contributions of all mesh triangles with respect to the rendered pixels. Such formulation enables our framework to flow gradients to the occluded and far-range vertices, which cannot be achieved by the previous state-of-the-arts. We show that by using the proposed renderer, one can achieve significant improvement in 3D unsupervised single-view reconstruction both qualitatively and quantitatively. Experiments also demonstrate that our approach is able to handle the challenging tasks in image-based shape fitting, which remain nontrivial to existing differentiable renderers.

Code Repositories

ShichenLiu/SoftRas
Official
pytorch
Mentioned in GitHub
kach/softraxterizer
jax
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
3d-object-reconstruction-on-shapenetSoftRas (full)
3DIoU: 0.6464
single-view-3d-reconstruction-on-shapenetSoftRas (full)
3DIoU: 0.6464
single-view-3d-reconstruction-on-shapenetvoxel [47]
3DIoU: 0.5736
single-view-3d-reconstruction-on-shapenetNMR [19]
3DIoU: 0.6015
single-view-3d-reconstruction-on-shapenetretrieval [47]
3DIoU: 0.4766

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