#include <opencv2/opencv.hpp>

#include <iostream>

using namespace cv;

using namespace std;

int main(int argc, char** argv) {

    Mat src = imread("D:/vcprojects/images/toux.jpg");

    if (src.empty()) {

        printf("could not load image...\n");

        return -;

    }

    namedWindow("input image", CV_WINDOW_AUTOSIZE);

    imshow("input image", src);

    Scalar colorTab[] = {

        Scalar(, , ),

        Scalar(, , ),

        Scalar(, , ),

        Scalar(, , ),

        Scalar(, , )

    };

    int width = src.cols;

    int height = src.rows;

    int dims = src.channels();

    // 像素点个数

    int sampleCount = width*height;

    int clusterCount = ;

    //将数据装载到一行

    Mat points(sampleCount, dims, CV_32F, Scalar());

    Mat labels;

    Mat centers(clusterCount, , points.type());

    // RGB 数据转换到样本数据

    int index = ;

    for (int row = ; row < height; row++) {

        for (int col = ; col < width; col++) {

            index = row*width + col;

            Vec3b bgr = src.at<Vec3b>(row, col);

            points.at<float>(index, ) = static_cast<int>(bgr[]);

            points.at<float>(index, ) = static_cast<int>(bgr[]);

            points.at<float>(index, ) = static_cast<int>(bgr[]);

        }

    }

    // 运行K-Means

    TermCriteria criteria = TermCriteria(TermCriteria::EPS + TermCriteria::COUNT, , 0.1);

    kmeans(points, clusterCount, labels, criteria, , KMEANS_PP_CENTERS, centers);

    // 显示图像分割结果

    Mat result = Mat::zeros(src.size(), src.type());

    for (int row = ; row < height; row++) {

        for (int col = ; col < width; col++) {

            index = row*width + col;

            int label = labels.at<int>(index, );

            result.at<Vec3b>(row, col)[] = colorTab[label][];

            result.at<Vec3b>(row, col)[] = colorTab[label][];

            result.at<Vec3b>(row, col)[] = colorTab[label][];

        }

    }

    for (int i = ; i < centers.rows; i++) {

        int x = centers.at<float>(i, );

        int y = centers.at<float>(i, );

        printf("center %d = c.x : %d, c.y : %d\n", i, x, y);

    }

    imshow("KMeans Image Segmentation Demo", result);

    waitKey();

    return ;

}