利用K-means聚类算法求出图像K个主色，实现对应区域的替换提取

背景：给定一幅彩色图片（RGB），利用K-means算法对图像中的颜色进行聚类，需要给定多少个主色。根据给定的主色计算图中不同主色与给定主色的偏差，给出最大偏差编号。将图中最大偏差替换为给定色彩

部分主要代码如下

clear all;
I = imread('E:\03.JPG');
[m, n, p] = size(I);
k = 4;%聚成四类
[main_co, label, J] = kmeans(I, k);
I_result= reshape(C(label, :), m, n, p);%main_co为主色调RGB值，I_result是聚类完的图片，只有四种颜色

function [C, label, J] = kmeans(I, k)
%C 是包含最终质心位置的的k*n矩阵矩阵。
[m, n, p] = size(I);
X = reshape(double(I), m*n, p);
rng('default');
C = X(randperm(m*n, k), :);
J_prev = inf; iter = 0; J = []; tol = 1e-2;
while true
    iter = iter + 1;
    dist = sum(X.^2, 2)*ones(1, k) + (sum(C.^2, 2)*ones(1, m*n))' - 2*X*C';
    [~, label] = min(dist, [], 2) ;
    for i = 1:k
       C(i, :) = mean(X(label == i , :));
    end
    J_cur = sum(sum((X - C(label, :)).^2, 2));
    J = [J, J_cur];
%     fprintf('#iteration: %03d, objective fcn: %f\n', iter, J_cur);
    if norm(J_cur-J_prev, 'fro') < tol
        break;
    end
    J_prev = J_cur;
end

参考链接：参考代码