swim/fold_slice
1
0
Fork 0
mirror of https://github.com/c-sooyoung/fold_slice.git synced 2026-09-18 01:59:11 +09:00
Code Issues Packages Projects Releases Wiki Activity

initial commit

Browse Source
This commit is contained in:
swim
2026-08-07 15:56:42 +09:00
commit 91ad25aca9
1012 changed files with 159314 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ptycho/+engines/+GPU/private/regulation_multilayers.m
+56
View File
@@ -0,0 +1,56 @@
% REGULATION_MULTILAYERS try to avoid ambiguity in the multilayer reconstruction by weakly forcing missing cone
% values towards zero
%
% self = regulation_multilayers(self, par, cache)
%
% ** self structure containing inputs: e.g. current reconstruction results, data, mask, positions, pixel size, ..
% ** par structure containing parameters for the engines
% ** cache structure with precalculated values to avoid unnecessary overhead
%
% returns:
% ++ self self-like structure with final reconstruction
%
function self = regulation_multilayers(self, par, cache)
import engines.GPU.GPU_wrapper.*
Npix = [self.Np_o, par.Nlayers];
for i = 1:3
grid{i} = ifftshift((-fix(Npix(i)/2):ceil(Npix(i)/2)-1))'/Npix(i);
grid{i} = shiftdim(grid{i},1-i);
end
% calculate force of regularization based on the idea that DoF = resolution^2/lambda
W = 1-atan(( par.regularize_layers * abs(grid{3}) ./ sqrt(grid{1}.^2+grid{2}.^2+1e-3)).^2) / (pi/2);
relax = 1;
alpha = 1;
Wa = W.*exp(-alpha*(grid{1}.^2 + grid{2}.^2));
for kk = 1:size(self.object,1)
obj = cat(3, self.object{kk,:});
% find correction for amplitude
aobj = abs(obj);
fobj = fftn(aobj);
fobj = fobj .* Wa;
aobj_upd = ifftn(fobj);
% push towards zero
aobj_upd = 1+0.9*(aobj_upd-1);
% find correction for phase
Wphase = min(1, 10*(cache.illum_sum_0{kk}/cache.MAX_ILLUM(kk)));
pobj = math.unwrap2D_fft2(obj,[],0,Wphase,-1);
fobj = (fftn((pobj)));
fobj = fobj .* Wa;
pobj_upd = ifftn(fobj);
obj_upd = Gfun(@regulation_multilayers_kernel,obj, aobj,aobj_upd, pobj, pobj_upd, Wphase, relax);
for ii = 1:par.Nlayers
self.object{kk,ii} = obj_upd(:,:,ii);
end
end
end
function [obj,corr] = regulation_multilayers_kernel(obj, aobj,aobj_upd, pobj, pobj_upd, weights, relax)
aobj_upd = (real(aobj_upd) - aobj);
pobj_upd = weights.*(real(pobj_upd) - pobj);
corr = (1+relax*aobj_upd) .* exp(1i*relax*pobj_upd);
obj = obj .* corr;
end