search

NVIDIAGameWorks / kaolin-wisp

NVIDIA Kaolin Wisp is a PyTorch library powered by NVIDIA Kaolin Core to work with neural fields (including NeRFs, NGLOD, instant-ngp and VQAD).
Other
1.45k stars 132 forks source link

Memory optimization patch in dataset initialization #164

Closed tovacinni closed 11 months ago

tovacinni commented 11 months ago
  1. Don't resize with cv2 if it's not needed
  2. Batch the CPU transfer