python - 是否有与 MATLAB 的 conv2 函数等效的 Python？

Question

Python 或其任何模块是否具有与 MATLAB 的 conv2 等效的功能？功能？更具体地说，我对与 MATLAB 中的 conv2(A, B, 'same') 进行相同计算的东西感兴趣。

Answer 1

虽然其他答案已经提到 scipy.signal.convolve2d 作为等效项，但我发现使用 mode='same' 时结果确实不同。

虽然 Matlab 的 conv2 会在图像的底部和右侧产生伪影，但 scipy.signal.convolve2d 在图像的顶部和左侧会产生相同的伪影。

查看这些链接以获取显示行为的图(没有足够的声誉直接发布图像):

Upper left corner of convoluted Barbara

Lower right corner of convoluted Barbara

以下包装器可能效率不高，但通过将输入数组和输出数组各旋转 180 度解决了我的问题:

import numpy as np
from scipy.signal import convolve2d

def conv2(x, y, mode='same'):
    return np.rot90(convolve2d(np.rot90(x, 2), np.rot90(y, 2), mode=mode), 2)