[1422d3]: / util / im2col_general.m

Download this file

21 lines (17 with data), 378 Bytes

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
function im = im2col_general(varargin)
%
NumInput = length(varargin);
InImg = varargin{1};
patchsize12 = varargin{2};
z = size(InImg,3);
im = cell(z,1);
if NumInput == 2
for i = 1:z
im{i} = im2colstep(InImg(:,:,i),patchsize12)';
end
else
for i = 1:z
im{i} = im2colstep(InImg(:,:,i),patchsize12,varargin{3})';
end
end
im = [im{:}]';