HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

TensoRF: Tensorial Radiance Fields

Anpei Chen; Zexiang Xu; Andreas Geiger; Jingyi Yu; Hao Su

TensoRF: Tensorial Radiance Fields

Abstract

We present TensoRF, a novel approach to model and reconstruct radiance fields. Unlike NeRF that purely uses MLPs, we model the radiance field of a scene as a 4D tensor, which represents a 3D voxel grid with per-voxel multi-channel features. Our central idea is to factorize the 4D scene tensor into multiple compact low-rank tensor components. We demonstrate that applying traditional CP decomposition -- that factorizes tensors into rank-one components with compact vectors -- in our framework leads to improvements over vanilla NeRF. To further boost performance, we introduce a novel vector-matrix (VM) decomposition that relaxes the low-rank constraints for two modes of a tensor and factorizes tensors into compact vector and matrix factors. Beyond superior rendering quality, our models with CP and VM decompositions lead to a significantly lower memory footprint in comparison to previous and concurrent works that directly optimize per-voxel features. Experimentally, we demonstrate that TensoRF with CP decomposition achieves fast reconstruction (<30 min) with better rendering quality and even a smaller model size (<4 MB) compared to NeRF. Moreover, TensoRF with VM decomposition further boosts rendering quality and outperforms previous state-of-the-art methods, while reducing the reconstruction time (<10 min) and retaining a compact model size (<75 MB).

Code Repositories

ashawkey/torch-ngp
pytorch
Mentioned in GitHub
apchenstu/TensoRF
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
low-dose-x-ray-ct-reconstruction-on-x3dTensoRF
PSNR: 33.78
SSIM: 0.9387
novel-view-synthesis-on-x3dTensoRF
PSNR: 37.67
SSIM: 0.9712

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