ncnn

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 "testutil.h"
16

17
static int test_instancenorm(const ncnn::Mat& a, float eps, int affine)
18
{
19
    int channels = a.c;
20

21
    ncnn::ParamDict pd;
22
    pd.set(0, affine ? channels : 0);
23
    pd.set(1, eps);
24
    pd.set(2, affine);
25

26
    std::vector<ncnn::Mat> weights(2);
27
    weights[0] = RandomMat(channels);
28
    weights[1] = RandomMat(channels);
29

30
    int ret = test_layer("InstanceNorm", pd, weights, a);
31
    if (ret != 0)
32
    {
33
        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);
34
    }
35

36
    return ret;
37
}
38

39
static int test_instancenorm_0()
40
{
41
    return 0
42
           || test_instancenorm(RandomMat(6, 4, 2), 0.01f, 0)
43
           || test_instancenorm(RandomMat(3, 3, 12), 0.002f, 0)
44
           || test_instancenorm(RandomMat(5, 7, 16), 0.02f, 0)
45
           || test_instancenorm(RandomMat(6, 4, 2), 0.01f, 1)
46
           || test_instancenorm(RandomMat(3, 3, 12), 0.002f, 1)
47
           || test_instancenorm(RandomMat(5, 7, 16), 0.02f, 1);
48
}
49

50
int main()
51
{
52
    SRAND(7767517);
53

54
    return 0
55
           || test_instancenorm_0();
56
}
57