/pytorch-noise

A pytorch wrapper for a modified version of https://github.com/covexp/cuda-noise

Primary LanguageCudaMIT LicenseMIT

pytorch-noise

A pytorch wrapper for a modified version of https://github.com/covexp/cuda-noise

Dependencies:

  • numpy
  • PyTorch (compiled with CUDA support)
  • cv2 (to run the examples)
  • a CUDA toolkit, to compile the CUDA/C++ extension.