OpenCV利用K-means實現根據顔色進行圖像分割

#include
#include


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

    cv::Mat img(500, 500, CV_8UC3);
    cv::RNG rng(12345);
    cv::Scalar colorTab[] = {
        cv::Scalar(0, 0, 255),
        cv::Scalar(0, 255, 0),
        cv::Scalar(255, 0, 0),
        cv::Scalar(0, 255, 255),
        cv::Scalar(255, 0, 255)
    };
    int numCluster = rng.uniform(2, 5);
    printf("種類數量 : %d\n", numCluster); 
    //4
    int sampleCount = rng.uniform(2, 1000);//隨機樣本
    printf("樣本數量 : %d\n", sampleCount);
    //591
    cv::Mat points(sampleCount, 1, CV_32FC2);
    cv::Mat labels;
    cv::Mat centers;

    for (int k = 0; k < numCluster; k++) {
        cv::Point center;
        center.x = rng.uniform(0, img.cols);//隨機坐標
        center.y = rng.uniform(0, img.rows);
        cv::Mat pointChunk = points.rowRange(k * sampleCount / numCluster, k == numCluster - 1 ? sampleCount : (k + 1) * sampleCount / numCluster);
        //每一類佔1/numCluster 行
        //rng.fill(pointChunk, cv::RNG::NORMAL, cv::Scalar(center.x, center.y), cv::Scalar(img.cols * 0.05, img.rows * 0.05));//用隨機數填充矩陣
        rng.fill(pointChunk, cv::RNG::UNIFORM, 0, 255);//用隨機數填充矩陣
        
    }

    randShuffle(points, 1, &rng);//算法打亂元素排列順序
    kmeans(points, numCluster, labels, cv::TermCriteria(cv::TermCriteria::EPS + cv::TermCriteria::COUNT, 10, 0.1), 3, cv::KMEANS_PP_CENTERS, centers);
    /*【根據蓡數1的坐標進行分類】
    蓡數1:爲cv::Mat類型，每行代表一個樣本，即特征，即mat.cols=特征長度，mat.rows=樣本數，數據類型僅支持float
    蓡數2:K  指定聚類時劃分爲幾類
    蓡數3:爲cv::Mat類型，是一個長度爲(樣本數，1)的矩陣，即mat.cols=1,mat.rows=樣本數；爲K-Means算法的結果輸出，指定每一個樣本聚類到哪一個label中【指定每一個樣本聚類到哪一類中】
    蓡數4:TermCriteria類，算法進行疊代時終止的條件，可以指定最大疊代次數，也可以指定預期的精度，也可以這兩種同時指定；
            蓡數1:int type 
                    type=TermCriteria::MAX_ITER/TermCriteria::COUNT  疊代到最大或疊代次數終止
                    type= TermCriteria::EPS   疊代到閾值終止
                    type= TermCriteria::MAX_ITER+ TermCriteria::EPS 上述兩者都作爲疊代終止條件
            蓡數2:int maxCount  疊代的最大次數
            蓡數3:double epsilon   閾值(中心位移值)
    蓡數5：指定K-Means算法執行的次數，每次算法執行的結果是不一樣的，選擇最好的那次結果輸出
    蓡數6:初始化均值點的方法，目前支持三種：KMEANS_RANDOM_CENTERS、KMEANS_PP_CENTERS、KMEANS_USE_INITIAL_LABELS 
    蓡數7:爲cv::Mat類型，輸出最終的均值點，mat.cols=特征長度，mat.rols=K 【每個類的中心點】

    */


    // 用不同顔色顯示分類
    //初始化圖片顔色。
    img = cv::Scalar::all(255);
    for (int i = 0; i < sampleCount; i++) {
        int index = labels.at(i);
        //獲取ponint點
        cv::Point p = points.at(i);
        //填充
        circle(img, p, 2, colorTab[index], -1, 8);
    }

    // 每個聚類的中心來繪制圓
    for (int i = 0; i < centers.rows; i++) {
        int x = centers.at(i, 0);
        int y = centers.at(i, 1);
        printf("c.x= %d, c.y=%d", x, y);
        circle(img, cv::Point(x, y), 40, colorTab[i], 1, cv::LINE_AA);
    }

    imshow("KMeans-Data-Demo", img);



        
    cv::waitKey(0);
    return 0;
}

#include
#include

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

    
    cv::Scalar colorTab[] = {
        cv::Scalar(0, 0, 255),
        cv::Scalar(0, 255, 0),
        cv::Scalar(255, 0, 0),
        cv::Scalar(0, 255, 255),
        
        };
    
    cv::Mat src = cv::imread("D:/bb/tu1/2.png");
    if (!src.data)
    {
        printf("圖像讀取失敗...\n");
        return -1;
    }
    cv::imshow("src", src);

    int width = src.cols;
    int height = src.rows;
    int dims = src.channels();

    int sampleCount = width * height;  //縂像素
    int clusterCount = 4;  //分類數量
    cv::Mat points(sampleCount, dims,CV_32F,cv::Scalar(10));
    //一個像素爲一行
    cv::Mat labels;
    cv::Mat centers(clusterCount,1, points.type());
    //保存中心坐標

    //把RGB數據轉換成樣本數據
    int index = 0;//像素序號
    for (int row = 0; row < height;row++) {
        for (int col = 0; col < width;col++) {
            index = row * width + col;
            cv::Vec3b bgr = src.at(row, col);
            points.at(index, 0) = static_cast(bgr[0]);
            points.at(index, 1) = static_cast(bgr[1]);
            points.at(index, 2) = static_cast(bgr[2]);

        }

    }

      
    cv::TermCriteria criteria = cv::TermCriteria(cv::TermCriteria::COUNT + cv::TermCriteria::EPS,10,0.1);
    //kmeans的終止的條件
    
    kmeans(points, clusterCount, labels, criteria, 3, cv::KMEANS_PP_CENTERS, centers);//運行kmeans 

    //顯示圖像分割結果
    cv::Mat result = cv::Mat::zeros(src.size(),src.type());
    for (int row = 0; row < height;row++) {
        for (int col = 0; col < width;col++) {
            index = row * width + col;
            int label = labels.at(index, 0); //獲取類序號
            result.at(row, col)[0] = colorTab[label][0];
            result.at(row, col)[1] = colorTab[label][1];
            result.at(row, col)[2] = colorTab[label][2];
            
        }
    }

    cv::imshow("result", result);





    cv::waitKey(0);
    return 0;
}