如何在numpy中进行卷积矩阵运算？

Question

有没有办法用numpy进行convolution matrix操作？

^{}只在1D数组上操作，因此这不是解决方案。

我宁愿避免使用scipy，因为在Windows上安装它似乎更困难。

Answer 1

我强烈建议为此使用openCV。但是原则上，您几乎可以直接使用wiki-article on kernel convolution上的“伪代码”来创建自己的函数。。。

ks = (kl-1)/2 ## kernels usually square with odd number of rows/columns
kl = len(kernel)
imx = len(matrix)
imy = len(matrix[0])
for i in range(imx):
  for j in range(imy):
    acc = 0
    for ki in range(kl): ##kernel is the matrix to be used
      for kj in range(kl):
        if 0 <= i-ks <= kl: ## make sure you don't get out of bound error
          acc = acc + (matrix[i-ks+ki][j-ks+kj] * kernel[ki][kj]) 
  matrix[i][j] = acc

原则上，这应该能做到（但我还没有测试过…）

我希望这对你有帮助。