Python进行常见图像形态学处理操作的示例代码

import numpy as np
import matplotlib.pyplot as plt
import scipy.ndimage as sn

x = np.zeros([20,20])
x[5:15, 5:15] = 1
x[10:12,10:12] = 0
x[2:4, 2:4] = 1

func = {
    "origin": lambda x : x,
    "erosion": sn.binary_erosion,
    "dilation": sn.binary_dilation,
    "opening": sn.binary_opening,
    "closing": sn.binary_closing
}

fig = plt.figure()
for i,key in enumerate(func):
    ax = fig.add_subplot(2,3,i+1)
    ax.imshow(func[key](x))
    plt.title(key)

plt.show()

from scipy.misc import ascent
img = ascent()

funcs = {
    "original": lambda x, tmp:x,
    "erosion" : sn.grey_erosion,
    "dilation" : sn.grey_dilation,
    "opening" : sn.grey_opening,
    "closing" : sn.grey_closing
}

fig = plt.figure()
for i, key in enumerate(funcs):
    ax = fig.add_subplot(2,5,i+1)
    plt.imshow(funcs[key](img, (10,10)), cmap=plt.cm.gray)
    plt.axis('off')
    plt.title(key)


ori = img[350:500, 400:500]
funcs = {
    "ori": lambda x, tmp:x,
    "white" : sn.white_tophat,
    "black" : sn.black_tophat,
    "gradient" : sn.morphological_gradient,
    "laplace" : sn.morphological_laplace,
}

for i, key in enumerate(funcs):
    ax = fig.add_subplot(2,5,i+6)
    plt.imshow(funcs[key](ori, (5,5)), cmap=plt.cm.gray)
    plt.title(key)
    plt.axis('off')

plt.show()