【图像分割】基于matlab RGB颜色分层图像分割【含Matlab源码 516期】

一、图像分割简介

理论知识参考：【基础教程】基于matlab图像处理图像分割【含Matlab源码 191期】

二、部分源代码

clear all; close all; clc;
A = double(imread('bird_small.tiff'));% 载入图片
dim = size(A,1); % 图片行数
k = 16; % 颜色分类的层数
means = zeros(k, 3); % Initialize means to randomly-selected colors in the original photo.
rand_x = ceil(dim*rand(k, 1));%初始means是k行k列随机数作为聚类中心
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));
            %repmat(A,k,1)对A矩阵进行k行的复制
            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
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);
       
        large_image(i,j,:) = means(ind,:);
    end 

  

 

  
1
  
2
  
3
  
4
  
5
  
6
  
7
  
8
  
9
  
10
  
11
  
12
  
13
  
14
  
15
  
16
  
17
  
18
  
19
  
20
  
21
  
22
  
23
  
24
  
25
  
26
  
27
  
28
  
29
  
30
  
31
  
32
  
33
  
34
  
35
  
36
  
37
  
38
  
39
  
40
  
41
  
42
  
43
  
44
  
45
  
46
  
47
  
48
  
49
  
50
  
51
 

三、运行结果

四、matlab版本及参考文献

1 matlab版本
2014a

2 参考文献
[1] 蔡利梅.MATLAB图像处理——理论、算法与实例分析[M].清华大学出版社，2020.
[2]杨丹,赵海滨,龙哲.MATLAB图像处理实例详解[M].清华大学出版社，2013.
[3]周品.MATLAB图像处理与图形用户界面设计[M].清华大学出版社，2013.
[4]刘成龙.精通MATLAB图像处理[M].清华大学出版社，2015.
[5]赵勇,方宗德,庞辉,王侃伟.基于量子粒子群优化算法的最小交叉熵多阈值图像分割[J].计算机应用研究. 2008,(04)

文章来源: qq912100926.blog.csdn.net，作者：海神之光，版权归原作者所有，如需转载，请联系作者。

原文链接：qq912100926.blog.csdn.net/article/details/114782884