Python进阶——二维图像卷积运算

最新推荐文章于 2023-12-19 14:24:27 发布

原创最新推荐文章于 2023-12-19 14:24:27 发布 · 2.9k 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#卷积 #python

python进阶专栏收录该内容

8 篇文章

订阅专栏

本文通过使用Python的NumPy和SciPy库，演示了如何进行二维图像的卷积运算。通过加载图像，定义Scharr算子，并应用信号处理库中的卷积函数，实现了图像边缘检测。同时展示了不同图像输入下的卷积效果。

部署运行你感兴趣的模型镜像

Python进阶——二维图像卷积运算

代码需求

实现二维图像的卷积运算

代码实现

import numpy as np
from scipy import signal
from scipy import misc
import matplotlib.pyplot as plt

face = misc.face(gray=True)   #face.dat
scharr = np.array([[ -3-3j, 0-10j,  +3 -3j],
                   [-10+0j, 0+ 0j, +10 +0j],
                   [ -3+3j, 0+10j,  +3 +3j]]) # Gx + j*Gy

grad = signal.convolve2d(face, scharr, boundary='symm', mode='same')
fig, (ax_orig, ax_mag) = plt.subplots(1, 2, figsize=(10, 6))
plt.rcParams['font.sans-serif']='SimHei'
ax_orig.imshow(face, cmap='gray')
ax_orig.set_title('first')
ax_orig.set_axis_off()

ax_mag.imshow(np.absolute(grad), cmap='gray')
ax_mag.set_title('second')
ax_mag.set_axis_off()

fig.show()