1
// Tencent is pleased to support the open source community by making ncnn available.
3
// Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
5
// Licensed under the BSD 3-Clause License (the "License"); you may not use this file except
6
// in compliance with the License. You may obtain a copy of the License at
8
// https://opensource.org/licenses/BSD-3-Clause
10
// Unless required by applicable law or agreed to in writing, software distributed
11
// under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
12
// CONDITIONS OF ANY KIND, either express or implied. See the License for the
13
// specific language governing permissions and limitations under the License.
17
static int test_deformableconv2d(int w, int h, int c, int outch, int kernel, int dilation, int stride, int pad, int bias)
19
const int kernel_extent_w = dilation * (kernel - 1) + 1;
20
const int kernel_extent_h = dilation * (kernel - 1) + 1;
21
const int out_w = (w + pad + pad - kernel_extent_w) / stride + 1;
22
const int out_h = (h + pad + pad - kernel_extent_h) / stride + 1;
23
std::vector<ncnn::Mat> a(3);
24
a[0] = RandomMat(w, h, c);
25
a[1] = RandomMat(out_w, out_h, kernel * kernel * 2);
26
a[2] = RandomMat(out_w, out_h, kernel * kernel);
35
pd.set(6, outch * c * kernel * kernel);
37
int activation_type = RAND() % 7; // 0 1 2 3 4 5 6
38
ncnn::Mat activation_params(2);
39
activation_params[0] = (activation_type == 6) ? RandomFloat(0, 1) : RandomFloat(-1, 0); // alpha
40
activation_params[1] = RandomFloat(0, 1); // beta
41
pd.set(9, activation_type);
42
pd.set(10, activation_params);
44
std::vector<ncnn::Mat> weights(bias ? 2 : 1);
45
weights[0] = RandomMat(outch * c * kernel * kernel);
47
weights[1] = RandomMat(outch);
49
float epsilon = 0.001;
50
int ret = test_layer("DeformableConv2D", pd, weights, a, 1, epsilon);
53
fprintf(stderr, "test_deformableconv2d failed w=%d h=%d c=%d outch=%d kernel=%d dilation=%d stride=%d pad=%d bias=%d act=%d actparams=[%f,%f]\n", w, h, c, outch, kernel, dilation, stride, pad, bias, activation_type, activation_params[0], activation_params[1]);
59
opt.use_packing_layout = true;
60
opt.use_fp16_packed = false;
61
opt.use_fp16_storage = false;
62
opt.use_fp16_arithmetic = false;
63
opt.use_bf16_storage = false;
64
opt.use_shader_pack8 = false;
65
opt.use_image_storage = false;
66
opt.use_sgemm_convolution = false;
67
opt.use_winograd_convolution = false;
69
ret = test_layer_opt("DeformableConv2D", pd, weights, opt, a, 1, epsilon);
72
fprintf(stderr, "test_deformableconv2d failed w=%d h=%d c=%d outch=%d kernel=%d dilation=%d stride=%d pad=%d bias=%d act=%d actparams=[%f,%f]\n", w, h, c, outch, kernel, dilation, stride, pad, bias, activation_type, activation_params[0], activation_params[1]);
79
opt.use_packing_layout = true;
80
opt.use_fp16_packed = true;
81
opt.use_fp16_storage = true;
82
opt.use_fp16_arithmetic = true;
83
opt.use_bf16_storage = true;
84
opt.use_shader_pack8 = true;
85
opt.use_image_storage = true;
86
opt.use_sgemm_convolution = false;
87
opt.use_winograd_convolution = false;
89
ret = test_layer_opt("DeformableConv2D", pd, weights, opt, a, 1, epsilon);
92
fprintf(stderr, "test_deformableconv2d failed w=%d h=%d c=%d outch=%d kernel=%d dilation=%d stride=%d pad=%d bias=%d act=%d actparams=[%f,%f]\n", w, h, c, outch, kernel, dilation, stride, pad, bias, activation_type, activation_params[0], activation_params[1]);
99
static int test_deformableconv2d_0()
101
static const int kdsp[10][4] = {
114
for (int i = 4; i < 6; i++)
116
const int k = kdsp[i][0];
117
const int d = kdsp[i][1];
118
const int s = kdsp[i][2];
119
const int p = kdsp[i][3];
122
|| test_deformableconv2d(9, 7, 1, 1, k, d, s, p, 1)
123
|| test_deformableconv2d(9, 7, 4, 13, k, d, s, p, 0)
124
|| test_deformableconv2d(9, 7, 13, 4, k, d, s, p, 1)
125
|| test_deformableconv2d(9, 7, 4, 8, k, d, s, p, 0)
126
|| test_deformableconv2d(9, 7, 8, 4, k, d, s, p, 1)
127
|| test_deformableconv2d(9, 7, 8, 13, k, d, s, p, 0)
128
|| test_deformableconv2d(9, 7, 13, 8, k, d, s, p, 1)
129
|| test_deformableconv2d(9, 7, 16, 16, k, d, s, p, 0)
130
|| test_deformableconv2d(16, 16, 1 * 3, 1 * 3, k, d, s, p, 1)
131
|| test_deformableconv2d(16, 16, 1 * 3, 4 * 3, k, d, s, p, 1)
132
|| test_deformableconv2d(16, 16, 1 * 3, 8 * 3, k, d, s, p, 1)
133
|| test_deformableconv2d(16, 16, 1 * 3, 16 * 3, k, d, s, p, 1)
134
|| test_deformableconv2d(16, 16, 4 * 3, 1 * 3, k, d, s, p, 1)
135
|| test_deformableconv2d(16, 16, 4 * 3, 4 * 3, k, d, s, p, 1)
136
|| test_deformableconv2d(16, 16, 4 * 3, 8 * 3, k, d, s, p, 1)
137
|| test_deformableconv2d(16, 16, 4 * 3, 16 * 3, k, d, s, p, 1)
138
|| test_deformableconv2d(16, 16, 8 * 3, 1 * 3, k, d, s, p, 1)
139
|| test_deformableconv2d(16, 16, 8 * 3, 4 * 3, k, d, s, p, 1)
140
|| test_deformableconv2d(16, 16, 8 * 3, 8 * 3, k, d, s, p, 1)
141
|| test_deformableconv2d(16, 16, 8 * 3, 16 * 3, k, d, s, p, 1)
142
|| test_deformableconv2d(16, 16, 16 * 3, 1 * 3, k, d, s, p, 1)
143
|| test_deformableconv2d(16, 16, 16 * 3, 4 * 3, k, d, s, p, 1)
144
|| test_deformableconv2d(16, 16, 16 * 3, 8 * 3, k, d, s, p, 1)
145
|| test_deformableconv2d(16, 16, 16 * 3, 16 * 3, k, d, s, p, 1);
158
return test_deformableconv2d_0();