1 // Tencent is pleased to support the open source community by making ncnn available.
2 //
3 // Copyright (C) 2020 THL A29 Limited, a Tencent company. All rights reserved.
4 //
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
7 //
8 // https://opensource.org/licenses/BSD-3-Clause
9 //
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.
14
15 #include "layer/instancenorm.h"
16 #include "testutil.h"
17
test_instancenorm(const ncnn::Mat & a,float eps,int affine)18 static int test_instancenorm(const ncnn::Mat& a, float eps, int affine)
19 {
20 int channels = a.c;
21
22 ncnn::ParamDict pd;
23 pd.set(0, affine ? channels : 0);
24 pd.set(1, eps);
25 pd.set(2, affine);
26
27 std::vector<ncnn::Mat> weights(2);
28 weights[0] = RandomMat(channels);
29 weights[1] = RandomMat(channels);
30
31 int ret = test_layer<ncnn::InstanceNorm>("InstanceNorm", pd, weights, a);
32 if (ret != 0)
33 {
34 fprintf(stderr, "test_instancenorm failed a.dims=%d a=(%d %d %d) eps=%f affine=%d\n", a.dims, a.w, a.h, a.c, eps, affine);
35 }
36
37 return ret;
38 }
39
test_instancenorm_0()40 static int test_instancenorm_0()
41 {
42 return 0
43 || test_instancenorm(RandomMat(6, 4, 2), 0.01f, 0)
44 || test_instancenorm(RandomMat(3, 3, 12), 0.002f, 0)
45 || test_instancenorm(RandomMat(5, 7, 16), 0.02f, 0)
46 || test_instancenorm(RandomMat(6, 4, 2), 0.01f, 1)
47 || test_instancenorm(RandomMat(3, 3, 12), 0.002f, 1)
48 || test_instancenorm(RandomMat(5, 7, 16), 0.02f, 1);
49 }
50
main()51 int main()
52 {
53 SRAND(7767517);
54
55 return 0
56 || test_instancenorm_0();
57 }
58