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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
%--------------------------------------------------------------------------
% This file is part of the ASTRA Toolbox
%
% Copyright: 2010-2021, imec Vision Lab, University of Antwerp
% 2014-2021, CWI, Amsterdam
% License: Open Source under GPLv3
% Contact: astra@astra-toolbox.com
% Website: http://www.astra-toolbox.com/
%--------------------------------------------------------------------------
classdef SmoothingGPU < matlab.mixin.Copyable
% Default policy class for smoothing for DART.
%----------------------------------------------------------------------
properties (Access=public)
radius = 1; % SETTING: Radius of smoothing kernel.
b = 0.1; % SETTING: Intensity of smoothing. Between 0 and 1.
full3d = 'yes'; % SETTING: smooth in 3D? {'yes','no'}
gpu_core = 0; % SETTING:
end
%----------------------------------------------------------------------
methods (Access=public)
%------------------------------------------------------------------
function settings = getsettings(this)
% Returns a structure containing all settings of this object.
% >> settings = DART.smoothing.getsettings();
settings.radius = this.radius;
settings.b = this.b;
settings.full3d = this.full3d;
end
%------------------------------------------------------------------
function V_out = apply(this, ~, V_in)
% Applies smoothing.
% >> V_out = DART.smoothing.apply(DART, V_in);
% 2D, one slice
if size(V_in,3) == 1
V_out = this.apply_2D(V_in);
% 3D, slice by slice
elseif ~strcmp(this.full3d,'yes')
V_out = zeros(size(V_in));
for slice = 1:size(V_in,3)
V_out(:,:,slice) = this.apply_2D(V_in(:,:,slice));
end
% 3D, full
else
V_out = this.apply_3D(V_in);
end
end
end
%----------------------------------------------------------------------
methods (Access=protected)
%------------------------------------------------------------------
function V_out = apply_2D(this, V_in)
vol_geom = astra_create_vol_geom(size(V_in));
in_id = astra_mex_data2d('create', '-vol', vol_geom, V_in);
out_id = astra_mex_data2d('create', '-vol', vol_geom, 0);
cfg = astra_struct('DARTSMOOTHING_CUDA');
cfg.InDataId = in_id;
cfg.OutDataId = out_id;
cfg.Intensity = this.b;
cfg.option.GPUindex = this.gpu_core;
alg_id = astra_mex_algorithm('create',cfg);
astra_mex_algorithm('iterate',alg_id,1);
V_out = astra_mex_data2d('get', out_id);
astra_mex_algorithm('delete', alg_id);
astra_mex_data2d('delete', in_id, out_id);
end
%------------------------------------------------------------------
function I_out = apply_3D(this, I_in)
r = this.radius;
w = 2 * r + 1;
% Set Kernel
K = ones(w,w,w) * this.b / (w.^3-1); % edges
K(r+1,r+1,r+1) = 1 - this.b; % center
% output window
I_out = zeros(size(I_in,1)+w-1, size(I_in,2)+w-1, size(I_in,3)+w-1);
% blur convolution
for s = -r:r
for t = -r:r
for u = -r:r
I_out(1+r+s:end-r+s, 1+r+t:end-r+t, 1+r+u:end-r+u) = I_out(1+r+s:end-r+s, 1+r+t:end-r+t, 1+r+u:end-r+u) + K(r+1+s, r+1+t, r+1+u) * I_in;
end
end
end
% shrink output window
I_out = I_out(1+r:end-r, 1+r:end-r, 1+r:end-r);
end
%------------------------------------------------------------------
end
end
|
