Kaolin加速 3D 深度學(xué)習(xí)研究

Kaolin 是一套用于加速 3D 深度學(xué)習(xí)研究的工具。

NVIDIA Kaolin 庫提供用于處理各種 3D 表征的 PyTorch API，并且包含日益增多的 GPU 優(yōu)化操作，例如模塊化可微渲染、表征之間快速轉(zhuǎn)換、數(shù)據(jù)加載、3D 檢查點(diǎn)等。

NVIDIA Omniverse Kaolin 應(yīng)用程序是一款面向 3D 深度學(xué)習(xí)研究人員的交互式應(yīng)用程序，允許檢查 3D 數(shù)據(jù)集、在訓(xùn)練期間交互式可視化模型的 3D 輸出，以及渲染合成數(shù)據(jù)集。該研究應(yīng)用程序基于 Omniverse 套件構(gòu)建，可使用高保真 RTX 渲染，并且定期通過新的擴(kuò)展程序獲取新功能。

同時，這些工具可以顯著縮短為各種 3D 應(yīng)用程序開發(fā) AI 研究所需的時間。

模塊化可微渲染器

借助模塊化和優(yōu)化的網(wǎng)格可微渲染器，開發(fā)先進(jìn)的反向圖形應(yīng)用程序。

3D 數(shù)據(jù)加載

輕松加載大型 3D 數(shù)據(jù)集，用于訓(xùn)練機(jī)器學(xué)習(xí)模型。使用 OBJ 和 USD 格式的導(dǎo)入和導(dǎo)出實(shí)用工具。

GPU 優(yōu)化的 3D 操作

借助快速可靠的轉(zhuǎn)換運(yùn)營，在 3D 表征之間轉(zhuǎn)換。使用 GPU 優(yōu)化的 3D 損失函數(shù)實(shí)現(xiàn)，以及日益增加的其他 3D 數(shù)據(jù)操作。

3D 檢查點(diǎn)

以 USD 格式導(dǎo)出網(wǎng)格、點(diǎn)云和立體像素網(wǎng)格的 3D 檢查點(diǎn)，實(shí)現(xiàn)模型訓(xùn)練的交互式可視化。