Applying Wiener filter to remove noise using python

user_8893369 0
Viewed 15395

This is the what my input image looks like

And this is what my wiener filtered output should look like

The first image is my input image. The second image is a wiener filtered image, which is my output.

Below is the code to use wiener filter on my image. The input image is "img5" and the output image is "Wiener_filtered".

psf = np.ones((5,5)) / 25
img6 = convolve2d(img5,psf,'same')
img6 += 0.1 * img6.std() * np.random.standard_normal(img6.shape)
Wiener_filtered = restoration.wiener(img6,psf,1100)

Below I have attached the input image "img5" and the outcomes of "img6" and "Wiener_filtered"

Input image img5

Input image "img5"

img6

Outcome of "img6"

Wiener_filtered

The final wiener filtered image

I need help in finding out where i went wrong. I am new to image processing. Can someone show me the correct method.

wiener-filter scikit-image opencv image-processing python-3.x
2 Answers
user_43529300

Try to use : img_as_float from skimage.util. I think it should work : 

from skimage.util import img_as_float

img5 = img_as_float(img5)
psf = np.ones((5,5)) / 25
img6 = convolve2d(img5,psf,'same')
img6 += 0.1 * img6.std() * np.random.standard_normal(img6.shape)
Wiener_filtered = restoration.wiener(img6,psf,1100)
user_79382560
Related