一段对图片按照rgb颜色聚类的python代码

一段对图片按照rgb颜色聚类的代码，
运行命令为：
python get_color_feature.py --image=./diffculty/342940_1_1.png
结果为显示3张图，分别是原图、剪切缩放后的图、取出聚类出的第二大类的mask

本代码时为了验证利用rgb（或者其他颜色空间聚类的效果），感兴趣的朋友可以试试


# import the necessary packages
from skimage.exposure import rescale_intensity
from skimage.segmentation import slic
from skimage.util import img_as_float
from skimage import io
import numpy as np
import argparse
import cv2
import scipy
import scipy.cluster.hierarchy as sch
import argparse

# 参数操作，
ap = argparse.ArgumentParser()
ap.add_argument("-i","--image",required=True,help="path to input image")
args = vars(ap.parse_args())

# 读入图片
oimage = cv2.imread(args["image"])

# 将图片缩放至[150,200]，降低聚类的复杂度，提高运行速度
orig = cv2.resize(oimage,(150,200),interpolation=cv2.INTER_CUBIC)

# 初始化显示模块
vis = np.zeros(orig.shape[:2],dtype="float")
# 定义图片剪切