python中图像的卷积

Question

我运行以下脚本 从PIL导入图像 将numpy作为np导入 从scipy导入信号作为sg

from PIL import Image
import numpy as np
from scipy import signal as sg

def np_from_img(fname):
return np.asarray(Image.open(fname), dtype=np.float32)

def save_as_img(ar, fname):
Image.fromarray(ar.round().astype(np.uint8)).save(fname)

def norm(ar):
return 255.*np.absolute(ar)/np.max(ar)

img="src/img.png"
save_as_img(norm(sg.convolve(img, [[1.],[-1.]])), 'src/portal-h.png')

我有以下错误： 值错误：in1和in2的维数应该相同

有人知道为什么吗？在

Answer 1

我猜你的输入是一个RBG图像，它被转换成3D数组而不是2D数组。在

尝试将img替换为img.mean(axis=-1)。在