举例原始slic算法的matlab代码

以下是一个简单的原始slic算法的matlab代码：

function [labels, numlabels] = slic(I, K, compactness)

% I: 输入图像
% K: 超像素数量
% compactness: 超像素紧密度参数

% 初始化
[height, width, ~] = size(I);
S = floor(sqrt(height*width/K));
numlabels = 0;
labels = zeros(height, width);

% 初始化聚类中心
cx = S/2:S:width-S/2;
cy = S/2:S:height-S/2;
[X, Y] = meshgrid(cx, cy);
cx = X(:);
cy = Y(:);
numclusters = length(cx);

% 迭代
while true
    numlabels = 0;
    for i = 1:numclusters
        xlow = round(cx(i)-S);
        xhigh = round(cx(i)+S);
        ylow = round(cy(i)-S);
        yhigh = round(cy(i)+S);
        if xlow < 1
            xlow = 1;
        end
        if xhigh > width
            xhigh = width;
        end
        if ylow < 1
            ylow = 1;
        end
        if yhigh > height
            yhigh = height;
        end
        subimage = I(ylow:yhigh, xlow:xhigh, :);
        [subheight, subwidth, ~] = size(subimage);
        subimage = reshape(subimage, subheight*subwidth, 3);
        sublabels = repmat(numlabels+1, subheight*subwidth, 1);
        D = pdist2(subimage, subimage);
        D = D/max(max(D));
        Dxy = pdist2([1:subheight]', [1:subwidth]');
        Dxy = Dxy/max(max(Dxy));
        D = D + compactness*Dxy;
        [temp, tempidx] = min(D);
        sublabels = sublabels + tempidx';
        labels(ylow:yhigh, xlow:xhigh) = sublabels;
        numlabels = numlabels + max(sublabels);
    end
    if numlabels == numclusters
        break;
    end
    cx = zeros(numlabels, 1);
    cy = zeros(numlabels, 1);
    for i = 1:numlabels
        [y, x] = find(labels == i);
        cx(i) = mean(x);
        cy(i) = mean(y);
    end
    numclusters = numlabels;
end

end

希望这个代码能够帮到你！