K-means应用于图像处理matlab程序

clear all; close all; clc; A = double(imread('bird_small.tiff'));% Load small image % A = double(imread('bird_large.tiff'));% Load large image dim = size(A,1); % number of pixels in picture's length/width k = 16; % number of colors to represent means = zeros(k, 3); % Initialize means to randomly-selected colors in the original photo. rand_x = ceil(dim*rand(k, 1)); rand_y = ceil(dim*rand(k, 1)); for i = 1:k means(i,:) = A(rand_x(i), rand_y(i), :); end for itr=1:100 s_x=zeros(k,3); s_ind=zeros(k,1); for i=1:dim for j=1:dim r=A(i,j,1);g=A(i,j,2);b=A(i,j,3); [val ind]=min(sum((repmat([r,g,b],k,1)-means).^2,2)); s_x(ind,:)=s_x(ind,:)+[r,g,b]; s_ind(ind)=s_ind(ind)+1; end end for ii=1:k if s_ind(ii)>0 s_x(ii,:)=s_x(ii,:)./s_ind(ii); end end d=sum(sqrt(sum((s_x-means).^2,2))) if d<1e-5 break end means=s_x; end means = round(means); itr figure; hold on for i=1:k col = (1/255).*means(i,:); rectangle('Position', [i, 0, 1, 1], 'FaceColor', col, 'EdgeColor', col); end axis off % Recalculate the big image and display large_image = double(imread('bird_large.tiff')); figure;imshow(uint8(round(large_image))); large_dim = size(large_image, 1); for i = 1:large_dim for j = 1:large_dim r = large_image(i,j,1); g = large_image(i,j,2); b = large_image(i,j,3); [val ind]=min(sum((repmat([r,g,b],k,1)-means).^2,2)); large_image(i,j,:) = means(ind,:); end end figure;imshow(uint8(round(large_image))); hold off imwrite(uint8(round(large_image)), 'bird_kmeans.jpg');% Save image