OpenCV k-means聚类

使用OpenCV对图像进行聚类

#include "opencv2/imgproc/imgproc.hpp"
#include "opencv2/highgui/highgui.hpp"

using namespace cv;

int main( int argc, char** argv )
{
  Mat src = imread( argv[1], 1 );
  Mat samples(src.rows * src.cols, 3, CV_32F);
  for( int y = 0; y < src.rows; y++ )
    for( int x = 0; x < src.cols; x++ )
      for( int z = 0; z < 3; z++)
        samples.at<float>(y + x*src.rows, z) = src.at<Vec3b>(y,x)[z];


  int clusterCount = 8;
  Mat labels;
  int attempts = 5;
  Mat centers;
  kmeans(samples, clusterCount, labels, TermCriteria(CV_TERMCRIT_ITER|CV_TERMCRIT_EPS, 0.0001, 10000), attempts, KMEANS_PP_CENTERS, centers );


  Mat new_image( src.size(), src.type() );
  for( int y = 0; y < src.rows; y++ )
    for( int x = 0; x < src.cols; x++ )
    {
      int cluster_idx = labels.at<int>(y + x*src.rows,0);
      new_image.at<Vec3b>(y,x)[0] = centers.at<float>(cluster_idx, 0);
      new_image.at<Vec3b>(y,x)[1] = centers.at<float>(cluster_idx, 1);
      new_image.at<Vec3b>(y,x)[2] = centers.at<float>(cluster_idx, 2);
    }
  imshow( "clustered image", new_image );
  waitKey( 0 );
}

对数据进行聚类

#include "opencv2/imgproc/imgproc.hpp"
#include "opencv2/highgui/highgui.hpp"
#include<iostream>
using namespace cv;
using namespace std;
int main( int argc, char** argv )
{
    float m[6][2]={{1,1},{1,2},{5,5},{5,6},{2,1},{6,5}};
    Mat samples=Mat(6,2,CV_32F,m); 
    int clusterCount=2;
    int attempts=5; 
    Mat centers; 
    Mat labels;
    kmeans(samples,clusterCount,labels,TermCriteria(CV_TERMCRIT_ITER|CV_TERMCRIT_EPS,0.0001,10000),attempts,KMEANS_PP_CENTERS,centers);
    int j=0;
    cout<<labels<<endl;
    cout<<centers<<endl;
} 

察看结果

[0; 0; 1; 1; 0; 1]
[1.3333334, 1.3333334;
  5.3333335, 5.3333335]

注意：

kmeans

聚类的矩阵类型，维度小于等于2，数据类型必须是CV_32F.