Matlab函数imfilter在Python中的对应

本文记录Matlab中的imfilter函数在Python下的对应函数,同时完美处理matlab卷积中对于边界的处理;文章引自ShuqiaoS博主,在此表示感谢,并加以验证和方法修改以及内容补充。

原文链接:https://blog.csdn.net/ShuqiaoS/article/details/110389141?spm=1001.2014.3001.5502

文章目录

首先给出结论:

Matlab Python
imfilter中的'conv' scipy.ndimage.convolve
imfilter中的'conv' scipy.ndimage.convolve
imfilter中的'corr' scipy.ndimage.correlate
imfilter中的'replicate'选项 上述函数中的mode='nearest'选项
n×n的核 相同的核
n×1的核 将数值放于n × n 的核的第n / 2 + 1 列(注意起始下标id=0),其余位置为0
1×n的核 将数值放于n × n 的核的第n / 2 行,其余位置为0

其中,可以用命令scipy.ndimage.filters.convolvescipy.ndimage.filters.correlate,也可以用scipy.ndimage.convolvescipy.ndimage.correlate,这两个命令是等价的

Conv

核尺寸为n×n时

下面的例子中,输入为:

核函数为:

Matlab命令:

python 复制代码
imfilter(img, kernel, 'conv', 'replicate')

对应Python命令:

python 复制代码
scipy.ndimage.filters.convolve(input, kernel, mode='nearest')

二者结果对比:

需要注意的是,上述命令中,Matlab下kernel的尺寸可以是列或行向量,但是Python下必须是一个n ∗ n 的矩阵,否则会报错。

核尺寸为n×1时

当核为列向量时,在Matlab中可以直接使用。比如当核为kernel = [1, 0, 1, 2, 1, 2, 3, 7, -1]时,Matlab命令:

python 复制代码
output_tmp = imfilter(tmp, kernel, 'conv', 'replicate');

的运行结果如下:

此时,如果想要得到相同的运算结果,则需要在Python下将值放置于中心位置,如下图所示:

此时,Python下的计算结果为:

与Matlab相同。

如果核尺寸为偶数,Python核有值的位置为n / 2 (Python中下标从0开始)。比如当n = 10 的时候,需要将第6列变成当前需要的数值(即对应Python下id=5)。例如,Matlab中核为kernel = [1, 0, 1, 2, 1, 2, 3, 7, -1, 1];,则Python下的核需要写成:

同样用命令

python 复制代码
output_tmp = imfilter(tmp, kernel, 'conv', 'replicate')  % Matlab命令
scipy.ndimage.filters.convolve(input, kernel, mode='nearest')  # Python命令

得到如下结果:

核尺寸为1×n时

当核为行向量时,在Matlab中可以直接使用,Python核有值的位置为n / 2 行。比如当核为kernel = [1, 0, 1, 2, 1, 2, 3, 7, -1]时,Matlab命令:

output_tmp = imfilter(tmp, kernel, 'conv', 'replicate')

的计算结果如下:

如果想得到同等对应的Python结果,则需要将核变成:

Python下的结果为

对于n 为偶数的情况也是相同的。如,Matlab下的核为kernel = [1, 0, 1, 2, 1, 2, 3, 7, -1, 1]时,命令

python 复制代码
output_tmp = imfilter(tmp, kernel, 'conv', 'replicate')

的结果为:

此时对应的Python核为:

结果为:

Corr

核尺寸为n×1时和核尺寸为1×n时的对应关系与前文Conv中所述相同,这里不再赘述。

下面的例子中,输入为:

核函数为:

Matlab命令:

python 复制代码
imfilter(img, kernel, 'corr', 'replicate')

对应Python命令:

python 复制代码
scipy.ndimage.filters.correlate(input, kernel, mode='nearest')


Conv2

虽然conv2在python中非常类似的替代函数convolve2d,但测试发现效果不好,会存在值不匹配问题,整体值上移一层,造成错误。

比如:

matlab代码:

python 复制代码
sstnum = conv2(sstnan,kernel,'same');

matlab结果为:

但使用python代码convolve2d方法:

python 复制代码
sstnum = convolve2d(sstnan1, kernel, mode='same', boundary='fill', fillvalue=0)

python结果为:

可以发现python的结果明显比matlab结果多了第一行,所以这种方法不可取。

推荐使用方法为:

python 复制代码
sstnum = convolve(sstnan1, np.rot90(kernel, 2), mode='constant', cval=0.0)

仍然使用卷积的方法,此时python结果为:

相关推荐
kaikaile199517 小时前
MATLAB实现自适应卡尔曼滤波(AKF)
开发语言·matlab
cici1587417 小时前
基于MATLAB的ADS-B接收机卫星与接收天线初始化实现
算法·matlab
程序员爱钓鱼17 小时前
Python编程实战 · 基础入门篇 | 变量与命名规范
后端·python
007php00717 小时前
猿辅导Java面试真实经历与深度总结(二)
java·开发语言·python·计算机网络·面试·职场和发展·golang
惊鸿.Jh18 小时前
C++可变参数模板
开发语言·python
MoRanzhi120318 小时前
Pillow 基础图像操作与数据预处理
图像处理·python·深度学习·机器学习·numpy·pillow·数据预处理
素素.陈18 小时前
向RAGFlow中上传文档到对应的知识库
开发语言·python
小宁爱Python18 小时前
Django Web 开发系列(一):视图基础与 URL 路由配置全解析
后端·python·django
茜茜西西CeCe18 小时前
数字图像处理-图像编码与压缩
人工智能·计算机视觉·matlab·数字图像处理·图像压缩·图像编码
空影星18 小时前
SiriKali,一款跨平台的加密文件管理器
python·编辑器·电脑·智能硬件