FilesExpand file tree

 master

/

denoise.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

executable file

·

25 lines (23 loc) · 939 Bytes

 master

/

denoise.py

Copy path

Top

File metadata and controls

• Code
• Blame

executable file

·

25 lines (23 loc) · 939 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

#!/usr/bin/env python

import matplotlib.pyplot as plt

import numpy as np

from scipy.signal import convolve2d

from skimage import color, data, restoration

if __name__ == '__main__':

image = color.rgb2gray(data.astronaut())

point_spread_func = np.ones((5, 5))/25.0

image = convolve2d(image, point_spread_func, 'same')

image += 0.1*image.std()*np.random.standard_normal(image.shape)

denoised, _ = restoration.unsupervised_wiener(image, point_spread_func)

figure, axes = plt.subplots(nrows=1, ncols=2, sharex=True, sharey=True,

subplot_kw={'adjustable': 'box-forced'},

figsize=(8,5))

plt.gray()

axes[0].imshow(image, vmin=denoised.min(), vmax=denoised.max())

axes[0].axis('off')

axes[0].set_title('original')

axes[1].imshow(denoised)

axes[1].axis('off')

axes[1].set_title('restored')

figure.tight_layout()

plt.show()