24
int Fold::load_param(const ParamDict& pd)
26
kernel_w = pd.get(1, 0);
27
kernel_h = pd.get(11, kernel_w);
28
dilation_w = pd.get(2, 1);
29
dilation_h = pd.get(12, dilation_w);
30
stride_w = pd.get(3, 1);
31
stride_h = pd.get(13, stride_w);
32
pad_left = pd.get(4, 0);
33
pad_right = pd.get(15, pad_left);
34
pad_top = pd.get(14, pad_left);
35
pad_bottom = pd.get(16, pad_top);
36
output_w = pd.get(20, 0);
37
output_h = pd.get(21, output_w);
42
int Fold::forward(const Mat& bottom_blob, Mat& top_blob, const Option& opt) const
44
const int max_channels = bottom_blob.h;
45
size_t elemsize = bottom_blob.elemsize;
47
const int kernel_extent_w = dilation_w * (kernel_w - 1) + 1;
48
const int kernel_extent_h = dilation_h * (kernel_h - 1) + 1;
50
const int outw = output_w + pad_left + pad_right;
51
const int outh = output_h + pad_top + pad_bottom;
53
const int inw = (outw - kernel_extent_w) / stride_w + 1;
54
const int inh = (outh - kernel_extent_h) / stride_h + 1;
58
const int maxk = kernel_w * kernel_h;
59
const int channels = max_channels / maxk;
61
Mat top_blob_bordered;
62
if (pad_left > 0 || pad_right > 0 || pad_top > 0 || pad_bottom > 0)
64
top_blob_bordered.create(outw, outh, channels, elemsize, opt.workspace_allocator);
68
top_blob_bordered = top_blob;
69
top_blob_bordered.create(outw, outh, channels, elemsize, opt.blob_allocator);
71
if (top_blob_bordered.empty())
75
const int gap = outw * stride_h - inw * stride_w;
77
#pragma omp parallel for num_threads(opt.num_threads)
78
for (int p = 0; p < channels; p++)
80
const float* sptr = bottom_blob.row(p * maxk);
81
Mat outm = top_blob_bordered.channel(p);
85
for (int u = 0; u < kernel_h; u++)
87
for (int v = 0; v < kernel_w; v++)
89
float* ptr = outm.row(dilation_h * u) + dilation_w * v;
91
for (int i = 0; i < inh; i++)
93
for (int j = 0; j < inw; j++)
107
if (pad_left > 0 || pad_right > 0 || pad_top > 0 || pad_bottom > 0)
110
opt_b.use_packing_layout = false;
111
copy_cut_border(top_blob_bordered, top_blob, pad_top, pad_bottom, pad_left, pad_right, opt_b);
112
if (top_blob.empty())
117
top_blob = top_blob_bordered;