gsplat: Uma Biblioteca de Código Aberto para Espalhamento Gaussiano

gsplat: An Open-Source Library for Gaussian Splatting

September 10, 2024

Autores: Vickie Ye, Ruilong Li, Justin Kerr, Matias Turkulainen, Brent Yi, Zhuoyang Pan, Otto Seiskari, Jianbo Ye, Jeffrey Hu, Matthew Tancik, Angjoo Kanazawa

cs.AI

Resumo

gsplat é uma biblioteca de código aberto projetada para treinar e desenvolver métodos de Splatting Gaussiano. Possui uma interface com Python bindings compatíveis com a biblioteca PyTorch e um núcleo com kernels CUDA altamente otimizados. gsplat oferece inúmeras funcionalidades que aprimoram a otimização de modelos de Splatting Gaussiano, incluindo melhorias de otimização para velocidade, memória e tempos de convergência. Resultados experimentais demonstram que gsplat alcança até 10% menos tempo de treinamento e 4 vezes menos memória do que a implementação original. Utilizado em vários projetos de pesquisa, gsplat é ativamente mantido no GitHub. O código-fonte está disponível em https://github.com/nerfstudio-project/gsplat sob a Licença Apache 2.0. Agradecemos contribuições da comunidade de código aberto.

English

gsplat is an open-source library designed for training and developing Gaussian Splatting methods. It features a front-end with Python bindings compatible with the PyTorch library and a back-end with highly optimized CUDA kernels. gsplat offers numerous features that enhance the optimization of Gaussian Splatting models, which include optimization improvements for speed, memory, and convergence times. Experimental results demonstrate that gsplat achieves up to 10% less training time and 4x less memory than the original implementation. Utilized in several research projects, gsplat is actively maintained on GitHub. Source code is available at https://github.com/nerfstudio-project/gsplat under Apache License 2.0. We welcome contributions from the open-source community.