MorphACWE图像分割

import cv2
import numpy as np
import matplotlib.pylab as pylab
from skimage.segmentation import (morphological_chan_vese,morphological_geodesic_active_contour,inverse_gaussian_gradient, checkerboard_level_set)
#定义存储迭代函数
def store_evolution_in(lst):
    def _store(x):
        lst.append(np.copy(x))
    return _store
img=cv2.imread('C:/Users/xpp/Desktop/Lena.png')#原始图像
gray=cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)#将彩色图片转换为灰度图片
init_lvl_set=checkerboard_level_set(gray.shape,6)#图像预处理
evolution=[]
callback=store_evolution_in(evolution)#存储迭代
lvl_set=morphological_chan_vese(gray,30,init_level_set=init_lvl_set,smoothing=3,iter_callback=callback)
fig,axes=pylab.subplots(2,1,figsize=(8,6))
axes=axes.flatten()
axes[0].imshow(img,cmap="gray"),axes[0].set_axis_off(),axes[0].contour(lvl_set,[0.5],colors='r')
axes[0].set_title("Morphological ACWE segmentation",fontsize=12)
axes[1].imshow(lvl_set,cmap="gray"),axes[1].set_axis_off()
contour=axes[1].contour(evolution[5],[0.5],colors='g')
contour.collections[0].set_label("Iteration 5")
contour=axes[1].contour(evolution[10],[0.5],colors='y')
contour.collections[0].set_label("Iteration 10")
contour=axes[1].contour(evolution[-1],[0.5],colors='r')
contour.collections[0].set_label("Iteration "+str(len(evolution)-1))
axes[1].legend(loc="upper right")
axes[1].set_title("Morphological ACWE evolution",fontsize=12)
fig.tight_layout()
pylab.show()