Files
scikit-image/doc/examples/plot_entropy2.py
T
2015-08-29 10:41:10 +02:00

39 lines
822 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()