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.46k stars 132 forks source link

Memory optimization patch in dataset initialization #164

Closed tovacinni closed 1 year ago

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