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