import numpy as np
from numpy.fft import fft2, ifft2
import cv2
from PIL import Image
def wiener_filter(img,kernel,K = 10):
kernel=([3,1],[2,1])
dummy = np.copy(img)
kernel = np.pad(kernel, [(0, dummy.shape[0] - kernel.shape[0]), (0, dummy.shape[1] - kernel.shape[1])], 'constant')
# Fourier Transform
dummy = fft2(dummy)
kernel = fft2(kernel)
kernel = np.conj(kernel) / (np.abs(kernel) ** 2 + K)
dummy = dummy * kernel
dummy = np.abs(ifft2(dummy))
return np.uint8(dummy)
img = cv2.imread("C:\\Users\\anup\\Desktop\\New folder\\leo.jpg")
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
#gray1 = cv2.cvtColor(gray,cv2.COLOR_GRAY2RGB)
kernel=([3,1],[2,1])
fm = wiener_filter(img,kernel)
cv2.imshow("Image", img)
key = cv2.waitKey(0)
我正在尝试实现维纳滤波器对模糊图像进行去卷积。我的实现是这样的
在实现这一点时,我遇到了一个错误
Traceback (most recent call last):
File "C:\Python27\deblur.py", line 21, in <module>
fm = wiener_filter(img,kernel)
File "C:\Python27\deblur.py", line 8, in wiener_filter
kernel = np.pad(kernel, [(0, dummy.shape[0] - kernel.shape[0]), (0, dummy.shape[1] - kernel.shape[1])], 'constant')
AttributeError: 'tuple' object has no attribute 'shape'
在那里:
您可以定义一个以
kernel
为参数的函数,然后直接覆盖它。你知道吗然后您尝试使用
kernel.shape
,这显然不存在于([3,1],[2,1])
显然,
shape
是用于numpy
ndarray
对象的方法。 所以也许这对你有用:或者像@Eric提到的:
相关问题 更多 >
编程相关推荐