MATLAB filter2/conv2 函数在 Python 语言中的等价函数

共计 432 个字符，预计需要花费 2 分钟才能阅读完成。

在 Python 中，可以使用 SciPy 库的 convolve2d 函数来实现 MATLAB 中 filter2 或conv2函数的功能。以下是一个示例代码：

import numpy as np
from scipy.signal import convolve2d
def filter2(mat, kernel):
return convolve2d(mat, kernel, mode='same')
# 示例用法
mat = np.array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
kernel = np.array([[1, 0, -1],
[2, 0, -2],
[1, 0, -1]])
result = filter2(mat, kernel)
print(result)

在上面的示例中，mat是输入矩阵，kernel是卷积核。filter2函数返回了卷积后的结果。可以通过 mode='same' 参数来保持输出与输入矩阵的大小相同，与 MATLAB 的 filter2 函数的默认行为相对应。

丸趣 TV 网 – 提供最优质的资源集合！