Files
scikit-image/doc/examples/plot_entropy2.py
T
odebeir f8afc07d80 pep8
2015-08-29 12:55:24 +02:00

39 lines
837 B
Python

"""
=======
Entropy
=======
The following example applies the local entropy measure to a noised image with a variable noise.
"""
import matplotlib.pyplot as plt
import numpy as np
from skimage.filters.rank import entropy
from skimage.morphology import disk
noise_mask = 28*np.ones((128, 128), dtype=np.uint8)
noise_mask[32:-32, 32:-32] = 30
noise = (noise_mask*np.random.random(noise_mask.shape)-.5*noise_mask).astype(np.uint8)
img = noise + 128
radius = 10
e = entropy(img, disk(radius))
plt.figure(figsize=[15, 5])
plt.subplot(1, 3, 1)
plt.imshow(noise_mask, cmap=plt.cm.gray)
plt.xlabel('noise mask')
plt.colorbar()
plt.subplot(1, 3, 2)
plt.imshow(img, cmap=plt.cm.gray)
plt.xlabel('noised image')
plt.colorbar()
plt.subplot(1, 3, 3)
plt.imshow(e)
plt.xlabel('image local entropy ($r=%d$)' % radius)
plt.colorbar()
plt.show()