http://blog.sina.com.cn/s/blog_5dd2e9270100xdcp.html
代码不能运行,不知道咋回事
poison matting matlab code 注:显示图像后,单击鼠标左键连出完全背景区域,双击左键结束; 图像改变后,再单击鼠标左键连出完全前景区域,双击左键结束。即可得到结果。 A=imread(‘ppmm2.bmp‘); imshow(A); hold on; Threshold = 1; UniBack=[0 0 255]; % choose background object, and B results into the Foreground Object and the % Boundary Region % choose the outer line of the boundary region. Single left click the mouse to specify vertice.Double left click or single right click to finish; OuterBW = roipoly(A); B(:,:,1) = immultiply(A(:,:,1),OuterBW); B(:,:,2) = immultiply(A(:,:,2),OuterBW); B(:,:,3) = immultiply(A(:,:,3),OuterBW); Back(:,:,1) = immultiply(A(:,:,1),~OuterBW); Back(:,:,2) = immultiply(A(:,:,2),~OuterBW); Back(:,:,3) = immultiply(A(:,:,3),~OuterBW); imshow(B); % choose foreground object, and C results into a Circle of Interest % choose the inner line of the boundary region. Single left click the mouse to specify vertice.Double left click or single right click to finish; InnerBW = roipoly(B); C(:,:,1) = immultiply(B(:,:,1),~InnerBW); C(:,:,2) = immultiply(B(:,:,2),~InnerBW); C(:,:,3) = immultiply(B(:,:,3),~InnerBW); Fore(:,:,1) = immultiply(A(:,:,1),InnerBW); Fore(:,:,2) = immultiply(A(:,:,2),InnerBW); Fore(:,:,3) = immultiply(A(:,:,3),InnerBW); imshow(C); hold off; RawAlpha = (double(OuterBW)+double(InnerBW))/2; I = double(A); % Method of "averaging" to get raw fore and background colors for i = 1:size(RawAlpha,1) for j = 1:size(RawAlpha,2) RawFore(i,j,1)=0; RawFore(i,j,2)=0; RawFore(i,j,3)=0; RawBack(i,j,1)=0; RawBack(i,j,2)=0; RawBack(i,j,3)=0; if RawAlpha(i,j) == 1 RawFore(i,j,:)=Fore(i,j,:); end; if RawAlpha(i,j) == 0 RawBack(i,j,:)=Back(i,j,:); end; if RawAlpha(i,j) == 0.5 r=1; while 1 TestFore = InnerBW(max(i-r,1):min(i+r,size(A,1)),max(j-r,1):min(j+r,size(A,2))); if size(find(TestFore),1) [i1, j1] = find(TestFore); i2 = i1 + max(i-r,1) -1; j2 = j1 + max(j-r,1) -1; Rs = double(Fore(i2,j2,:)); Ds(:,1)=diag(Rs(:,:,1)); Ds(:,2)=diag(Rs(:,:,2)); Ds(:,3)=diag(Rs(:,:,3)); if size(Ds,1) == 1 RawFore(i,j,:) =Ds; else RawFore(i,j,:) = sum(Ds) / size(Ds,1); end; clear Ds; break; else r=r+1; end; end; r=1; while 1 TestBack = ~OuterBW(max(i-r,1):min(i+r,size(A,1)),max(j-r,1):min(j+r,size(A,2))); if size(find(TestBack),1) [i1, j1] = find(TestBack); i2 = i1 + max(i-r,1) -1; j2 = j1 + max(j-r,1) -1; Rs = double(Back(i2,j2,:)); Ds(:,1)=diag(Rs(:,:,1)); Ds(:,2)=diag(Rs(:,:,2)); Ds(:,3)=diag(Rs(:,:,3)); if size(Ds,1) == 1 RawBack(i,j,:) =Ds; else RawBack(i,j,:) = sum(Ds) / size(Ds,1); end; clear Ds; break; else r=r+1; end; end; end; end; end; %此处没有用高斯滤波 Denorm = RawFore - RawBack; %red channel I1 = I(:,:,1); Denorm1 = Denorm(:,:,1); for i=1:size(Denorm1,1) for j= 1: size(Denorm1,2) if Denorm1(i,j)==0 Denorm1(i,j)=1; end; end; end; OldAlpha = RawAlpha; NewAlpha = RawAlpha; h1=0; while 1 for i=1:size(OldAlpha,1) for j=1:size(OldAlpha,2) NewAlpha(i,j) = OldAlpha(i,j); if RawAlpha(i,j) == 0.5 Roui = ((I1(i+1,j) + I1(i-1,j) - 2 * I1(i,j)) * Denorm1(i,j) - (I1(i+1,j) - I1(i,j)) * (Denorm1(i+1,j) - Denorm1(i,j)))/(Denorm1(i,j) * Denorm1(i,j)); Rouj = ((I1(i,j+1) + I1(i,j-1) - 2 * I1(i,j)) * Denorm1(i,j) - (I1(i,j+1) - I1(i,j)) * (Denorm1(i,j+1) - Denorm1(i,j)))/(Denorm1(i,j) * Denorm1(i,j)); Rou = Roui + Rouj; NewAlpha(i,j) = (OldAlpha(i+1,j) + NewAlpha(i-1,j) + OldAlpha(i,j+1) + NewAlpha(i,j-1) - Rou) / 4; if NewAlpha(i,j)<0 NewAlpha(i,j)=0; end; if NewAlpha(i,j)>1 NewAlpha(i,j)=1; end; end; end; end; % imshow(uint8(NewAlpha*255)); DifferenceAlpha = abs(NewAlpha - OldAlpha); OldAlpha = NewAlpha; if sum(sum(DifferenceAlpha)) < Threshold break; end; h1=h1+1; end; for i=1:size(A,1) for j=1:size(A,2) if OldAlpha(i,j)==0 NewImage(i,j,:)=UniBack‘; else NewImage(i,j,1)=UniBack(1)*(1-OldAlpha(i,j))+RawFore(i,j,1)*OldAlpha(i,j); NewImage(i,j,2)=UniBack(2)*(1-OldAlpha(i,j))+RawFore(i,j,2)*OldAlpha(i,j); NewImage(i,j,3)=UniBack(3)*(1-OldAlpha(i,j))+RawFore(i,j,3)*OldAlpha(i,j); end; end; end; figure,imshow(NewAlpha); figure,imshow(uint8(NewImage)); figure,imshow(uint8(RawFore)); figure,imshow(uint8(RawBack));
时间: 2024-10-13 19:45:08