Kaolin：一个用于加速 3D 深度学习研究的 PyTorch 库
计算机视觉与模式识别
2019-11-14 v2 机器学习
机器人学
摘要
我们提出 Kaolin，一个旨在加速 3D 深度学习研究的 PyTorch 库。Kaolin 为深度学习系统提供了可微 3D 模块的高效实现。凭借加载和预处理多个流行 3D 数据集的功能，以及操作网格、点云、有符号距离函数和体素网格的原生函数，Kaolin 减少了对浪费性样板代码编写的需求。Kaolin 打包了多个可微图形模块，包括渲染、光照、着色和视角扭曲。Kaolin 还支持一系列损失函数和评估指标以实现无缝评估，并提供可视化功能来渲染 3D 结果。重要的是，我们策划了一个包含许多最先进 3D 深度学习架构的综合模型库，作为未来研究工作的起点。Kaolin 作为开源软件可在 https://github.com/NVIDIAGameWorks/kaolin/ 获取。
引用
@article{arxiv.1911.05063,
title = {Kaolin: A PyTorch Library for Accelerating 3D Deep Learning Research},
author = {Krishna Murthy Jatavallabhula and Edward Smith and Jean-Francois Lafleche and Clement Fuji Tsang and Artem Rozantsev and Wenzheng Chen and Tommy Xiang and Rev Lebaredian and Sanja Fidler},
journal= {arXiv preprint arXiv:1911.05063},
year = {2019}
}
备注
Kaolin is available as open-source software at https://github.com/NVIDIAGameWorks/kaolin/