sqdist.m

来自「利用matlab实现彩色图像的分割。算法主要是利用聚类算法。」· M 代码 · 共 23 行

function d = sqdist(mtx, mean)

% mtx is MxNx3, mean is 1x3, d is MxN

running = mtx;

% find squared diff from mean of each color component
running(:,:,1) = (running(:,:,1)-mean(1)).^2;
running(:,:,2) = (running(:,:,2)-mean(2)).^2;
running(:,:,3) = (running(:,:,3)-mean(3)).^2;

% do sum for each pixel
d = sum(running, 3);


% testing - output should be 1
%
% testim=rand(4,4,3);
% testmu = [0.2 0.6 0.4];
% d=sqdist(testim, testmu);
% sum((reshape(testim(3,3,:),1,3)-testmu).^2) == d(3,3)