利用随机游走算法进行图像分割

利用随机游走算法进行图像分割

from skimage import  io,img_as_ubyte,img_as_float
from skimage.color import rgb2gray
import matplotlib.pyplot as plt
import numpy as np

# 读取图像
img = img_as_float(io.imread("macro-photography-of-strawberry-934066.jpg"))
gray_img = rgb2gray(img)
h,w = gray_img.shape

# 查看直方图
plt.hist(gray_img.flat,bins =100,range=(0,1))

# 随机游走分割
markers = np.zeros_like(gray_img)
markers[gray_img>0.8]=1
markers[gray_img<0.45]=2
from skimage.segmentation import random_walker
# 使用随机游走算法获得的结果
labels = random_walker(gray_img,markers,beta=10,mode="bf")
# plt.imshow(labels)

# 转换成布尔类型
segm1 = (labels<1.1)
# print(segm1)

# 形态学开运算操作
from skimage.morphology import opening,disk
kernel = disk(10)   
img_opening = opening(segm1,kernel)
print(type(img_opening))

# 将单通道阈值，转为RGB通道的阈值
segm = np.tile(img_opening.reshape(h,w,1),3)
# print(segm)

# 复制彩色图像
rgb_img = img.copy()
# 掩膜操作
rgb_img[segm] = 0

# 显示图像
plt.figure(figsize=(10,8),dpi=80)
plt.subplot(121)
plt.imshow(img)
plt.xlabel("原图",fontproperties='SimHei')
plt.subplot(122)
plt.imshow(rgb_img)
plt.xlabel("分割结果",fontproperties='SimHei')
plt.show()

Figure_1.png
randomwalk.png

参考网站：
skimage.segmentation中random_walker