/kaolin

A PyTorch Library for Accelerating 3D Deep Learning Research

Primary LanguagePythonApache License 2.0Apache-2.0

Kaolin: A Pytorch Library for Accelerating 3D Deep Learning Research

Overview

NVIDIA Kaolin library provides a PyTorch API for working with a variety of 3D representations and includes a growing collection of GPU-optimized operations such as modular differentiable rendering, fast conversions between representations, data loading, 3D checkpoints and more.

Kaolin library is part of a larger suite of tools for 3D deep learning research. For example, the Omniverse Kaolin App will allow interactive visualization of 3D checkpoints. To find out more about the Kaolin ecosystem, visit the NVIDIA Kaolin Dev Zone page.

Installation and Getting Started

Visit the Kaolin Library Documentation to get started!

About the Latest Release (0.10.0)

With the version 0.10.0 we are focusing on Volumetric rendering, adding new features for tetrahedral meshes, including DefTet volumetric renderer and losses, and Deep Marching Tetrahedrons, and adding new primitive operations for efficient volumetric rendering of Structured Point Clouds, we are also adding support for materials with USD importation.

Finally we are adding two new tutorials to show how to use the latest features from Kaolin:

  • How to use DMtet to rencontruct a mesh from point clouds generated by the Omniverse Kaolin App

DMtet Tutorial

SPC Tutorial

See change logs for details.

Contributing

Please review our contribution guidelines.

External Projects using Kaolin

Citation

If you found this codebase useful in your research, please cite

@misc{Kaolin2019,
author = {Clement Fuji Tsang and Masha Shugrina and Jean Francois Lafleche and Towaki Takikawa and Jiehan Wang and Charles Loop and Wenzheng Chen and Krishna Murthy Jatavallabhula and Edward Smith and Artem Rozantsev and Or Perel and Frank Shen and Jun Gao and Sanja Fidler and Gavriel State and Jason Gorski and Tommy Xiang and Jianing Li and Michael Li and Rev Lebaredian}
title = {Kaolin}
year = {2019}
}

Contributors

Current Team:

  • Technical Lead: Clement Fuji Tsang
  • Manager: Masha Shugrina
  • Jean-Francois Lafleche
  • Charles Loop
  • Or Perel
  • Towaki Takikawa
  • Jiehan Wang

Other Majors Contributors:

  • Wenzheng Chen
  • Sanja Fidler
  • Jun Gao
  • Jason Gorski
  • Rev Lebaredian
  • Jianing Li
  • Michael Li
  • Krishna Murthy Jatavallabhula
  • Artem Rozantsev
  • Frank Shen
  • Edward Smith
  • Gavriel State
  • Tommy Xiang