summaryrefslogtreecommitdiffstats
path: root/SD-VBS/common/toolbox/MultiNcut/computeMultiW.m
diff options
context:
space:
mode:
Diffstat (limited to 'SD-VBS/common/toolbox/MultiNcut/computeMultiW.m')
-rwxr-xr-xSD-VBS/common/toolbox/MultiNcut/computeMultiW.m245
1 files changed, 245 insertions, 0 deletions
diff --git a/SD-VBS/common/toolbox/MultiNcut/computeMultiW.m b/SD-VBS/common/toolbox/MultiNcut/computeMultiW.m
new file mode 100755
index 0000000..b2cb7ea
--- /dev/null
+++ b/SD-VBS/common/toolbox/MultiNcut/computeMultiW.m
@@ -0,0 +1,245 @@
1%inputs : image, number of layers, distance defining the subgrid, the edge filter scales for each layer, radius for each layer,
2%edge variance for filter, shape of the neighbourhood layout ('square', 'star', 'cross'), sigma for intensity affinity,
3% sigma for distance influence in affinity, weight coefficients for Wpps in the multiscale matrix.
4%output : multiscale affinity matrix , extern constraint matrix, affinity matrices of each layer seperatly.
5
6
7function [multiWpp,constraintMat, Wind,data,emag,ephase]= computeMultiW (image,data);
8
9%variables
10
11if isempty(data.layers.number)
12 n=2;
13else
14 n=data.layers.number;
15end
16
17if isempty(data.layers.dist)
18 dist_grid=3;
19else
20 dist_grid=data.layers.dist;
21end
22
23if isempty(data.W.scales)
24 s=1:n;
25elseif (length(data.W.scales)==n)
26 s=data.W.scales;
27else
28 s=1:n;
29end
30
31if isempty(data.W.radius)
32 r(1)=2;
33 for i=2:n
34 r(i)=10;
35 end
36else
37 r=data.W.radius;
38end
39
40
41if isempty(data.W.edgeVariance)
42 data.W.edgeVariance=0.1;
43end
44
45if isempty(data.W.gridtype)
46 data.W.gridtype='square';
47end
48
49if isempty(data.W.sigmaI)
50 data.W.sigmaI=0.12;
51end
52
53if isempty(data.W.sigmaX)
54 data.W.sigmaX=10;
55end
56
57if isempty(data.layers.weight)
58 coef(1)=5;
59 coef(2:n)=200;
60elseif (length(data.layers.weight)==n)
61 coef=data.layers.weight;
62else
63 coef(1)=5;
64 coef(2:n)=100; %200
65end
66
67if isempty(data.W.mode)
68 data.W.mode=mixed;
69end
70
71
72[p1,q1,ignore]=size(image);
73image=image(:,:,1);
74filter_par = [4,30,4]; %[9,30,4]
75[x,y,gx,gy,par,threshold,emag,ephase,g,FIe,FIo,mago] = quadedgep2(image,filter_par,data,0.001);
76minW=10^(-2); %-3
77
78
79% function [multiWpp,constraintMat,p,q,Wppp,subgrid] = computemultiWpp(n,imageX,r,dist_grid,s,dataWpp,emag,ephase,minW,mode,facteurMul,contrainte,tt,gridtype,colormode,imageOriginale,subgridImageReduite,pG,qG)
80
81p= p1*ones(n,1);
82q= q1*ones(n,1);
83d= dist_grid*ones(n,1);
84d(1)=1;
85for (i=2:n)
86 d(i)=d(i)*3^(i-2);
87end
88p=ceil(p1./d);
89q=ceil(q1./d);
90
91%computation of the subgrids (the first pixel is coded by one). S{i,j}(k) is the index of
92%the kth pixel of the jth grid in the ith grid.
93
94for i=1:n-1
95 for j=i+1:n
96 a=[0:p(j)*q(j)-1];
97 subgrid{i,j}=p(i)*(floor(a/p(j)))*d(j)/d(i)+(1+mod(a,p(j))*d(j)/d(i));
98 end
99end
100
101%computation of the independent W matrix for each layer Wind{i} 1=<i=<n.
102
103[w1i,w1j]=cimgnbmap([p1,q1], r(1), 1);
104
105if strcmp(data.W.mode,'mixed')
106 rMin = 0;
107 imageXX=double(image(:));
108 sigmaI= (std(imageXX(:)) + 1e-10 )* data.W.sigmaI;
109 Wpp{1} = multiIntensityFirstLayer(double(image),w1i,w1j,rMin,data.W.sigmaX,sigmaI,minW);
110 Wpp2= affinityic(emag(:,:,s(1)),ephase(:,:,s(1)),w1i,w1j,max(max(emag(:,:,s(1)))) * data.W.edgeVariance);
111 Wpp{1} = sqrt(Wpp{1} .* Wpp2)+0.1*Wpp2;
112
113elseif strcmp(data.W.mode,'notmixed')
114 Wpp{1}= affinityic(emag(:,:,s(1)),ephase(:,:,s(1)),w1i,w1j,max(max(emag(:,:,s(1)))) * data.W.edgeVariance);
115
116elseif strcmp(data.W.mode,'intensity')
117 rMin = 0;
118 imageXX=double(image(:));
119 sigmaI= (std(imageXX(:)) + 1e-10 )* data.W.sigmaI;
120 Wpp{1} = multiIntensityFirstLayer(double(image),w1i,w1j,rMin,data.W.sigmaX,sigmaI,minW);
121
122end
123Wpp{1}=coef(1)*(Wpp{1}+Wpp{1}')/2;
124%Wpp{1}= coef(1)*Wpp{1};
125Wind{1}=Wpp{1};
126
127
128
129
130for i=2:n
131 if strcmp(data.W.gridtype,'square')
132 [wwi,wwj]=cimgnbmap([p(i),q(i)], r(i), 1);
133 elseif strcmp(data.W.gridtype,'star')
134 [wwi,wwj]=cimgnbmap_star([p(i),q(i)], r(i), 1);
135 elseif strcmp(data.W.gridtype,'cross')
136 [wwi,wwj]=cimgnbmap_cross([p(i),q(i)], r(i), 1);
137 end
138 wwi=double(wwi);
139 wiInOriginalImage=(p1*(floor(wwi/p(i)))*d(i))+(1+mod(wwi,p(i))*d(i));
140 wiInOriginalImage=(p1*(floor(wwi/p(i)))*d(i))+(1+mod(wwi,p(i))*d(i));
141
142 wiInOriginalImage= uint32(wiInOriginalImage);
143
144if strcmp(data.W.mode,'mixed')
145 Wpp2= multiAffinityic(emag(:,:,i),ephase(:,:,i),wiInOriginalImage,wwj,subgrid{1,i},p(i),q(i),uint32(wwi),max(max(emag(:,:,i))) * data.W.edgeVariance);
146 a=floor(d(i)/d(i-1));
147 if (mod(a,2)==0)
148 a=a+1;
149 end
150% Wpp{i} = multiIntensityWppc(double(imageX),wiInOriginalImage,wwj,rMin,dataWpp.sigmaX,sigmaI,minW,subgrid{1,i},p(i),q(i),wi{i});
151
152 Wpp{i} = multiIntensityWppc(double(image),wiInOriginalImage,wwj,rMin,data.W.sigmaX,sigmaI,minW,subgrid{1,i},p(i),q(i),uint32(wwi));
153 Wpp{i} = sqrt(Wpp{i} .* Wpp2)+0.1*Wpp2;
154elseif strcmp(data.W.mode,'notmixed')
155 Wpp{i}= multiAffinityic(emag(:,:,i),ephase(:,:,i),wiInOriginalImage,wwj,subgrid{1,i},p(i),q(i),uint32(wwi),max(max(emag(:,:,i))) * data.W.edgeVariance);
156elseif strcmp(data.W.mode,'intensity')
157 Wpp{i} = multiIntensityWppc(double(image),wiInOriginalImage,wwj,rMin,data.W.sigmaX,sigmaI,minW,subgrid{1,i},p(i),q(i),uint32(wwi));
158end
159Wpp{i}= coef(i)*(Wpp{i}+Wpp{i}')/2;
160
161%Wpp{i}= coef(i)*Wpp{i};
162Wind{i}=Wpp{i};
163
164end
165
166%computation of the intern contraint matrices C{i,j}.
167
168for i=1:n-1
169 r=floor(d(i+1)/(d(i)*2));
170 [wwi,wwj]=cimgnbmap([p(i),q(i)], r, 1);
171 wi{i}=wwi;
172 wj{i}=wwj;
173end
174
175for i=1:n-1
176 for j=i+1:n
177 C{i,j}=sparse(p(i)*q(i),p(j)*q(j));
178 firstPointer=double(wj{i}(subgrid{i,j}))+1;
179 lastPointer=double(wj{i}(subgrid{i,j}+1));
180 invNbNeighbours=1./(lastPointer-firstPointer+1);
181 for (k=1:p(j)*q(j))
182 for (m=firstPointer(k):lastPointer(k))
183 C{i,j}(double(wi{i}(m))+1,k)=invNbNeighbours(k);
184 end
185 end
186 end
187end
188
189%Assembling the built matrices to make up multiWpp.
190for i=1:n
191 if (i>1)
192 for j=i-1:-1:1
193