地图着色程序

gaohaikuo1

1. function main()
   
2. I = imread('abc1.jpg');
   
3. I =I(1:300,1:300,:);
   
4. subplot(1,3,1);imshow(I);title('raw');
   
5. I_gray = rgb2gray(I);
   
6. %subplot(3,4,2);imshow(I_gray);title('gray');
   
7. I_background=medfilt2(I_gray ,[20,20]);
   
8. %subplot(3,4,4);imshow(I_background);
   
9. I_signal = I_gray -I_background;
   
10. %subplot(3,4,5);imshow(I_signal);title('remove noise');
    
11. I_bw = I_signal >0;
    
12. %subplot(3,4,6);imshow(I_bw );title('binary value');
    
13. I_bw1 = bwareaopen(I_bw,30,4) ;
    
14. %subplot(3,4,7);imshow(I_bw1);title('remove island');
    
15. I_bw2=  imdilate(I_bw1, strel('disk',1) );
    
16. %subplot(3,4,8);imshow(I_bw2,[]);title('dilate edge');
    
17. [I_label,num] = bwlabel( bwareaopen(1-I_bw2,30,4));
    
18. %subplot(3,4,9);imshow(I_label,[]);title('labeled');
    
19. I_label1 =  imdilate2noEdge(I_label);
    
20. %subplot(3,4,10);imshow(I_label1,[]);title('labeled1');
    
21. M = markMap(I_label1);
    
22. fprintf('M构建完成');
    
23. ColorMap = getColorMap(M);
    
24. fprintf('ColorMap构建完成');
    
25. M_Colored = getMColor(M,ColorMap);
    
26. colors = [255,0,0;0,255,0;0,0,255;255,255,0;255,0,255;0,255,255];
    
27. I_colored = zeros(size(I));
    
28. for i = 1:num
    
29.     I_colored(:,:,1) = I_colored(:,:,1) + (I_label1==i)*colors(ColorMap(i),1);
    
30.     I_colored(:,:,2) = I_colored(:,:,2) + (I_label1==i)*colors(ColorMap(i),2);
    
31.     I_colored(:,:,3) = I_colored(:,:,3) + (I_label1==i)*colors(ColorMap(i),3);
    
32.     fprintf('原图伪彩色构建%d/%d\n',[i,num]);
    
33. end
    
34. subplot(1,3,2);imshow(I_colored);title('colored');
    
35. [area,ratio] = getSizeDistribute(I_label1)
    
36. subplot(1,3,3);bar(area,ratio*100);
    
37. xlabel('area(um^2)');xlabel('ratio(%)');
    
38. 
    
39. end
    
40. 
    
41. 
    
42. 
    
43. function I_label1 =  imdilate2noEdge(I_label)
    
44. [m,n] = size(I_label);
    
45. flag  =1;
    
46. I_label1 = I_label;
    
47. while flag
    
48.     flag = 0;
    
49.     for i = 2:m-1
    
50.         for j=2:n-1
    
51.             if I_label(i,j) ==0
    
52.                 if I_label(i-1,j)~=0
    
53.                     I_label1(i,j) = I_label(i-1,j);flag  =1;
    
54.                 elseif I_label(i+1,j)~=0
    
55.                     I_label1(i,j) = I_label(i+1,j);flag  =1;
    
56.                 elseif I_label(i,j-1)~=0
    
57.                     I_label1(i,j) = I_label(i,j-1);flag  =1;
    
58.                 elseif I_label(i,j+1)~=0
    
59.                     I_label1(i,j) = I_label(i,j+1);flag  =1;
    
60.                 end
    
61.             end
    
62.         end
    
63.     end
    
64.     I_label = I_label1;
    
65. end
    
66. I_label1 = I_label;
    
67. end
    
68. function M = markMap(I_label)
    
69. num = max(max(I_label));
    
70. M = [(1:num)',zeros(num,30)];
    
71. 
    
72. [m,n] = size(I_label);
    
73. for i = 2:m-1
    
74.     for j=2:n-1
    
75.         indexs = [i-1,j;i,j-1];
    
76.         for i_index = 1:length(indexs(:,1))
    
77.              if I_label(indexs(i_index,1),indexs(i_index,2))~=I_label(i,j)
    
78.                  num1 = I_label(indexs(i_index,1),indexs(i_index,2));
    
79.                  num2 = I_label(i,j);
    
80.                  if num1~=0 && num2~=0
    
81.                      M = addNewToM(M ,num1,num2);
    
82.                      M = addNewToM(M ,num2,num1);  
    
83.                  end
    
84.              end
    
85.         end
    
86.     end
    
87. end
    
88. end
    
89. 
    
90. 
    
91. function M = addNewToM(M ,num1,num2)
    
92.      jj = 1;
    
93.      while M(num1,jj)~=0 && M(num1,jj)~=num2
    
94.          jj=jj+1;
    
95.      end
    
96.      M(num1,jj) = num2;
    
97. end
    
98. 
    
99. function ColorMap = getColorMap(M)
    
100. num = length(M(:,1));
     
101. ColorMap = zeros(num,1);
     
102. %1先选择1作为#1初始颜色
     
103. ColorMap(1) = 1;
     
104. ii = 1;
     
105. while find(ColorMap == 0)
     
106.     ColorMap = addOneColor(M,ColorMap);   
     
107.     fprintf('ColorMap构建%d/%d\n',[ii,num]); ii=ii+1;
     
108. end
     
109. end
     
110. 
     
111. function ColorMap = addOneColor(M,ColorMap)
     
112. %找出第一个未标记颜色，并且周围标记颜色最多的区域#2
     
113. num = length(M(:,1));
     
114. M_colored = zeros(size(M));
     
115. for i = 1:num
     
116.     if ColorMap(i) ~=0
     
117.              M_colored = M_colored+(M==i);
     
118.     end
     
119. end
     
120. num_colored = sum(M_colored(:,2:end),2).*(~M_colored(:,1));
     
121. [C,i_max] = max(num_colored);
     
122. %给区域#2标记颜色
     
123. i_color = 1;
     
124. ii = 2;
     
125. cc=[];
     
126. while M(i_max,ii)~=0
     
127.     cc = [cc,ColorMap(M(i_max,ii))];
     
128.     ii=ii+1;
     
129. end
     
130. color_target = 1;
     
131. while ~isempty(find(cc==color_target))
     
132.     color_target = color_target+1;
     
133. end
     
134. ColorMap(i_max) = color_target;
     
135. end
     
136. 
     
137. 
     
138. function M_colored = getMColor(M,ColorMap)
     
139. %找出第一个未标记颜色，并且周围标记颜色最多的区域#2
     
140. num = length(M(:,1));
     
141. M_colored = zeros(size(M));
     
142. for i = 1:num
     
143.     if ColorMap(i) ~=0
     
144.              M_colored = M_colored+ ColorMap(i)*(M==i);
     
145.     end
     
146. end
     
147. end
     
148. 
     
149. function [area,ratio] = getSizeDistribute(I_label)
     
150.     num = max(max(I_label));
     
151.     sizes = [];
     
152.     for i = 1:num
     
153.          sizes = [sizes,sum(sum(I_label==i))];
     
154.     end
     
155.     sizes = sizes/54.76;%change pixel2 to um2
     
156.     minv = min(sizes);
     
157.     maxv = max(sizes);
     
158.     d= 10;
     
159.     area = (maxv-minv)/d*(0:d) + minv;
     
160.     number = [];
     
161.     for i = 1:d
     
162.         number=[number,sum(sum(sizes>area(i) & sizes<area(i+1)))];
     
163.     end
     
164.     ratio = number/sum(number);
     
165.     area = area(2:end);
     
166. end
     
167. 
     
168. 
     
169. 
     
170. 
     
171. 
     

复制代码