@Fu Xianjun.All Rights Reserved.
@[TOC]图像分割与提取
需要的图片:
代码如下:
import cv2
import numpy as np
img = cv2.imread("coins.jpg")#获取原图
gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)#灰度图转换
#------二值化-----#
ret,binary=cv2.threshold(gray,0,255,cv2.THRESH_BINARY_INV+cv2.THRESH_OTSU)#阈值处理(反转)
#-----形态学-------#
#开操作
kernel =cv2.getStructuringElement(cv2.MORPH_RECT,(3,3))
opening = cv2.morphologyEx(binary,cv2.MORPH_OPEN,kernel,iterations=2)
sure_bg=cv2.dilate(opening,kernel,iterations=3)#膨胀。确认前景
sure_fg=cv2.erode(opening,kernel,iterations=3)#腐蚀。确认后景
unknow= cv2.subtract(sure_bg,sure_fg)#膨胀-腐蚀
#-----距离变化-------#
dist_transform=cv2.distanceTransform(opening,cv2.DIST_L2,5)
cv2.normalize(dist_transform,dist_transform,0,1.0,cv2.NORM_MINMAX)
#------二值化-----#
ret,sure_fg=cv2.threshold(dist_transform,0.5*dist_transform.max(),255,0)#阈值处理
sure_fg=np.uint8(sure_fg)#转换成8字节
unknow= cv2.subtract(sure_bg,sure_fg)#膨胀-腐蚀
ret,markers=cv2.connectedComponents(sure_fg)
markers = makers+1
markers[unknow==255]=0
#应用分水岭算法
markers=cv2.watershed(img,markers)
img[markers==-1]=[0,0,255]
cv2.imshow("img",img)#显示图片
cv2.imshow("b",binary)#显示图片
cv2.imshow("opening",opening)#显示图片
cv2.imshow("sure_bg",sure_bg)#显示图片
cv2.imshow("sure_fg",sure_fg)#显示图片
cv2.imshow("unknow",unknow)#显示图片
cv2.imshow("markers",markers_copy)#显示图片
cv2.waitKey(0)
cv2.destroyAllWindows()#释放窗口
运行结果: