算法源于西瓜书309页介绍的约束种子k均值算法
1.事先标注好一部分数据:
绿色部分是前景 对应的rgb是[0,255,0]
蓝色部分是背景 rgb[0,0,255]
2.
根据标注好的数据,遍历原始图片中的每一个像素点
计算该像素点与前景、背景的距离
与前景距离小就并入前景,否则并入背景
循环往复,直至前景,背景不再变化
代码:
import matplotlib.image as mpimg
import numpy as np
from PIL import Image
im = mpimg.imread('D:\\pythonProject\\data\\selectedimages\\monkey.bmp')
im2 = mpimg.imread('D:\\pythonProject\\data\\selectedimages\\monkey_marker.bmp')
im