我正在寻找一些实现 3d 卷积的源代码。理想情况下,我需要 C++ 代码或 CUDA 代码。如果有人能给我指点一个又好又快的实现,我将不胜感激:-)
干杯
最佳答案
您了解卷积通常是通过使用 fft 来完成的吗?参见,例如,http://en.wikipedia.org/wiki/Convolution
所以你需要一个 fft 库。
Fastest method to compute convolution建议 http://www.fftw.org/ (对于传统 CPU)。
对于 cuda,使用 cufft - http://www.gsic.titech.ac.jp/~ccwww/tebiki/tesla_e/tesla6_e.html
关于c++ - C++ 中的 3d 卷积,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/1948895/