为什么它说在我已经宣布克恩

2024-04-26 17:59:43 发布

您现在位置:Python中文网/ 问答频道 /正文

    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'

Tags: fromimageimportnumpyimgnpfiltercv2
1条回答
网友
1楼 · 发布于 2024-04-26 17:59:43

在那里:

def wiener_filter(img,kernel,K = 10):
    kernel=([3,1],[2,1])

您可以定义一个以kernel为参数的函数,然后直接覆盖它。你知道吗

然后您尝试使用kernel.shape,这显然不存在于([3,1],[2,1])

显然,shape是用于numpyndarray对象的方法。 所以也许这对你有用:

def wiener_filter(img,kernel,K = 10):
    kernel=np.array([[3,1],[2,1]])

或者像@Eric提到的:

def wiener_filter(img,kernel,K = 10):
    kernel=np.array(kernel)

相关问题 更多 >