1. SqueezeMe: Efficient Gaussian Avatars for VR
- Author
-
Saito, Shunsuke, Pidhorskyi, Stanislav, Santesteban, Igor, Iandola, Forrest, Gupta, Divam, Pahuja, Anuj, Bartolovic, Nemanja, Yu, Frank, Garbin, Emanuel, and Simon, Tomas
- Subjects
Computer Science - Computer Vision and Pattern Recognition - Abstract
Gaussian Splatting has enabled real-time 3D human avatars with unprecedented levels of visual quality. While previous methods require a desktop GPU for real-time inference of a single avatar, we aim to squeeze multiple Gaussian avatars onto a portable virtual reality headset with real-time drivable inference. We begin by training a previous work, Animatable Gaussians, on a high quality dataset captured with 512 cameras. The Gaussians are animated by controlling base set of Gaussians with linear blend skinning (LBS) motion and then further adjusting the Gaussians with a neural network decoder to correct their appearance. When deploying the model on a Meta Quest 3 VR headset, we find two major computational bottlenecks: the decoder and the rendering. To accelerate the decoder, we train the Gaussians in UV-space instead of pixel-space, and we distill the decoder to a single neural network layer. Further, we discover that neighborhoods of Gaussians can share a single corrective from the decoder, which provides an additional speedup. To accelerate the rendering, we develop a custom pipeline in Vulkan that runs on the mobile GPU. Putting it all together, we run 3 Gaussian avatars concurrently at 72 FPS on a VR headset. Demo videos are at https://forresti.github.io/squeezeme., Comment: v2
- Published
- 2024