opencv学习笔记五十四:KMeans图像分割

Kmeans算法是非常经典的聚类算法,在数据挖掘中kmeans经常用来做数据预处理,在图像处理中也可以用作图像的分割。

原理:首先选择要聚成几个类,然后初始化中心点。比如说要聚成3类,就随机生成3个中心点,然后每个数据点选择离它最近的中心点作为一类,对组成的类取平均值就会得到三个新的中心点,再对其进行聚类,不断迭代,当迭代的次数超过某一阈值或者前后两次中心点位置不变了或变化很小就终止聚类。

kmeans( InputArray data, int K, InputOutputArray bestLabels, TermCriteria criteria, int attempts, int flags, OutputArray centers = noArray() );
  • data表示用于聚类的数据点,是N维的数组类型(Mat型),必须浮点型;
  • K表示需要聚类的类别数;
  • bestLabels聚类后的标签数组,Mat型;
  • criteria迭代收敛准则(MAX_ITER最大迭代次数,EPS最高精度);
  • attemps表示尝试的次数,防止陷入局部最优;
  • flags 表示聚类中心的选取方式(KMEANS_RANDOM_CENTERS 随机选取,KMEANS_PP_CENTERS使用Arthur提供的算法,KMEANS_USE_INITIAL_LABELS使用初始标签);
  • centers 表示聚类后的类别中心;
#include
using namespace cv;

int main(int arc, char** argv) { 
	Mat src = imread("1.jpg");
	namedWindow("input", CV_WINDOW_AUTOSIZE);
	imshow("input", src

你可能感兴趣的:(opencv)