|
- // Copyright 2020 Tencent
- // SPDX-License-Identifier: BSD-3-Clause
-
- #include "testutil.h"
-
- static std::vector<int> IntArray(int a0)
- {
- std::vector<int> m(1);
- m[0] = a0;
- return m;
- }
-
- static std::vector<int> IntArray(int a0, int a1)
- {
- std::vector<int> m(2);
- m[0] = a0;
- m[1] = a1;
- return m;
- }
-
- static std::vector<int> IntArray(int a0, int a1, int a2)
- {
- std::vector<int> m(3);
- m[0] = a0;
- m[1] = a1;
- m[2] = a2;
- return m;
- }
-
- static std::vector<int> IntArray(int a0, int a1, int a2, int a3)
- {
- std::vector<int> m(4);
- m[0] = a0;
- m[1] = a1;
- m[2] = a2;
- m[3] = a3;
- return m;
- }
-
- static void print_int_array(const std::vector<int>& a)
- {
- fprintf(stderr, "[");
- for (size_t i = 0; i < a.size(); i++)
- {
- fprintf(stderr, " %d", a[i]);
- }
- fprintf(stderr, " ]");
- }
-
- static int test_crop(const ncnn::Mat& a, const std::vector<int>& starts_array, const std::vector<int>& ends_array, const std::vector<int>& axes_array)
- {
- ncnn::Mat starts(starts_array.size());
- {
- int* p = starts;
- for (size_t i = 0; i < starts_array.size(); i++)
- {
- p[i] = starts_array[i];
- }
- }
-
- ncnn::Mat ends(ends_array.size());
- {
- int* p = ends;
- for (size_t i = 0; i < ends_array.size(); i++)
- {
- p[i] = ends_array[i];
- }
- }
-
- ncnn::Mat axes(axes_array.size());
- {
- int* p = axes;
- for (size_t i = 0; i < axes_array.size(); i++)
- {
- p[i] = axes_array[i];
- }
- }
-
- ncnn::ParamDict pd;
- pd.set(9, starts); // starts
- pd.set(10, ends); // ends
- pd.set(11, axes); // axes
-
- std::vector<ncnn::Mat> weights(0);
-
- int ret = test_layer("Crop", pd, weights, a);
- if (ret != 0)
- {
- fprintf(stderr, "test_crop failed a.dims=%d a=(%d %d %d %d)", a.dims, a.w, a.h, a.d, a.c);
- fprintf(stderr, " starts=");
- print_int_array(starts_array);
- fprintf(stderr, " ends=");
- print_int_array(ends_array);
- fprintf(stderr, " axes=");
- print_int_array(axes_array);
- fprintf(stderr, "\n");
- }
-
- return ret;
- }
-
- static int test_crop_1d(const ncnn::Mat& a)
- {
- std::vector<int> params[][3] = {
- {IntArray(12), IntArray(-233), IntArray(0)},
- {IntArray(16), IntArray(-233), IntArray(0)},
- {IntArray(11), IntArray(11 + 16), IntArray(0)},
- {IntArray(12), IntArray(12 + 7), IntArray(-1)},
- {IntArray(16), IntArray(16 + 12), std::vector<int>()},
- {IntArray(11), IntArray(-7 + 1), IntArray(0)},
- {IntArray(12), IntArray(-12 + 1), IntArray(-1)},
- {IntArray(16), IntArray(-16 + 1), std::vector<int>()}
- };
-
- for (int i = 0; i < sizeof(params) / sizeof(params[0]); i++)
- {
- int ret = test_crop(a, params[i][0], params[i][1], params[i][2]);
- if (ret)
- return ret;
- }
-
- return 0;
- }
-
- static int test_crop_2d(const ncnn::Mat& a)
- {
- std::vector<int> params[][3] = {
- {IntArray(12), IntArray(-233), IntArray(0)},
- {IntArray(8), IntArray(-233), IntArray(0)},
- {IntArray(4), IntArray(-233), IntArray(1)},
- {IntArray(5, 11), IntArray(-233, -233), IntArray(0, 1)},
- {IntArray(11), IntArray(11 + 16), IntArray(0)},
- {IntArray(12), IntArray(12 + 7), IntArray(0)},
- {IntArray(8), IntArray(8 + 12), IntArray(-2)},
- {IntArray(5), IntArray(8), IntArray(1)},
- {IntArray(6), IntArray(9), IntArray(1)},
- {IntArray(4), IntArray(12), IntArray(-1)},
- {IntArray(11, 5), IntArray(11 + 7, 11), IntArray(0, 1)},
- {IntArray(12, 6), IntArray(12 + 12, 12), IntArray(0, 1)},
- {IntArray(8, 4), IntArray(8 + 16, 10), IntArray(0, -1)},
- {IntArray(11), IntArray(-16 + 1), IntArray(0)},
- {IntArray(12), IntArray(-7 + 1), IntArray(0)},
- {IntArray(8), IntArray(-12 + 1), IntArray(-2)},
- {IntArray(5), IntArray(-5 + 1), IntArray(1)},
- {IntArray(6), IntArray(-6 + 1), IntArray(1)},
- {IntArray(4), IntArray(-4 + 1), IntArray(-1)},
- {IntArray(11, 5), IntArray(-12 + 1, -6 + 1), IntArray(0, 1)},
- {IntArray(12, 6), IntArray(-16 + 1, -5 + 1), IntArray(0, 1)},
- {IntArray(8, 4), IntArray(-7 + 1, -4 + 1), IntArray(-2, -1)}
- };
-
- for (int i = 0; i < sizeof(params) / sizeof(params[0]); i++)
- {
- int ret = test_crop(a, params[i][0], params[i][1], params[i][2]);
- if (ret)
- return ret;
- }
-
- return 0;
- }
-
- static int test_crop_3d(const ncnn::Mat& a)
- {
- std::vector<int> params[][3] = {
- {IntArray(11), IntArray(-233), IntArray(0)},
- {IntArray(8), IntArray(-233), IntArray(0)},
- {IntArray(5), IntArray(-233), IntArray(1)},
- {IntArray(6), IntArray(-233), IntArray(2)},
- {IntArray(4), IntArray(-233), IntArray(-1)},
- {IntArray(12, 6), IntArray(-233, -233), IntArray(0, 1)},
- {IntArray(11, 5), IntArray(-233, -233), IntArray(0, -1)},
- {IntArray(8, 4), IntArray(-233, -233), IntArray(0, 2)},
- {IntArray(6, 6), IntArray(-233, -233), IntArray(1, -1)},
- {IntArray(11, 5, 5), IntArray(-233, -233, -233), IntArray(0, 1, 2)},
- {IntArray(8, 4, 4), IntArray(-233, -233, -233), IntArray(0, 1, -1)},
- {IntArray(11), IntArray(11 + 7), IntArray(0)},
- {IntArray(12), IntArray(12 + 12), IntArray(0)},
- {IntArray(8), IntArray(8 + 16), IntArray(0)},
- {IntArray(5), IntArray(13), IntArray(1)},
- {IntArray(6), IntArray(12), IntArray(1)},
- {IntArray(4), IntArray(11), IntArray(-2)},
- {IntArray(5), IntArray(12), IntArray(2)},
- {IntArray(6), IntArray(11), IntArray(2)},
- {IntArray(4), IntArray(13), IntArray(-1)},
- {IntArray(11, 5), IntArray(11 + 7, 11), IntArray(0, 1)},
- {IntArray(12, 6), IntArray(12 + 16, 12), IntArray(0, 1)},
- {IntArray(8, 4), IntArray(8 + 12, 13), IntArray(0, -2)},
- {IntArray(11, 5), IntArray(11 + 16, 13), IntArray(0, 2)},
- {IntArray(12, 6), IntArray(12 + 12, 11), IntArray(0, 2)},
- {IntArray(8, 4), IntArray(8 + 7, 12), IntArray(0, -1)},
- {IntArray(5, 4), IntArray(12, 12), IntArray(1, 2)},
- {IntArray(6, 3), IntArray(13, 13), IntArray(1, 2)},
- {IntArray(4, 2), IntArray(11, 11), IntArray(-2, -1)},
- {IntArray(11, 5, 2), IntArray(11 + 7, 11, 11), IntArray(0, 1, 2)},
- {IntArray(12, 6, 4), IntArray(12 + 16, 12, 12), IntArray(0, 1, 2)},
- {IntArray(8, 4, 3), IntArray(8 + 12, 13, 13), IntArray(-3, -2, -1)},
- {IntArray(11), IntArray(-7 + 1), IntArray(0)},
- {IntArray(12), IntArray(-12 + 1), IntArray(0)},
- {IntArray(8), IntArray(-16 + 1), IntArray(-3)},
- {IntArray(5), IntArray(-6 + 1), IntArray(1)},
- {IntArray(6), IntArray(-5 + 1), IntArray(1)},
- {IntArray(4), IntArray(-4 + 1), IntArray(-2)},
- {IntArray(5), IntArray(-5 + 1), IntArray(2)},
- {IntArray(6), IntArray(-4 + 1), IntArray(2)},
- {IntArray(4), IntArray(-6 + 1), IntArray(-1)},
- {IntArray(11, 5), IntArray(-7 + 1, -4 + 1), IntArray(0, 1)},
- {IntArray(12, 6), IntArray(-12 + 1, -6 + 1), IntArray(0, 1)},
- {IntArray(8, 4), IntArray(-16 + 1, -5 + 1), IntArray(-3, -2)},
- {IntArray(11, 5), IntArray(-12 + 1, -6 + 1), IntArray(0, 2)},
- {IntArray(12, 6), IntArray(-16 + 1, -5 + 1), IntArray(0, 2)},
- {IntArray(8, 4), IntArray(-7 + 1, -4 + 1), IntArray(-3, -1)},
- {IntArray(5, 2), IntArray(-5 + 1, -5 + 1), IntArray(1, 2)},
- {IntArray(6, 4), IntArray(-4 + 1, -4 + 1), IntArray(1, 2)},
- {IntArray(4, 3), IntArray(-6 + 1, -6 + 1), IntArray(-2, -1)},
- {IntArray(11, 5, 4), IntArray(-7 + 1, -5 + 1, -5 + 1), IntArray(0, 1, 2)},
- {IntArray(12, 6, 3), IntArray(-12 + 1, -6 + 1, -6 + 1), IntArray(0, 1, 2)},
- {IntArray(8, 4, 2), IntArray(-16 + 1, -4 + 1, -4 + 1), IntArray(-3, -2, -1)}
- };
-
- for (int i = 0; i < sizeof(params) / sizeof(params[0]); i++)
- {
- int ret = test_crop(a, params[i][0], params[i][1], params[i][2]);
- if (ret)
- return ret;
- }
-
- return 0;
- }
-
- static int test_crop_4d(const ncnn::Mat& a)
- {
- std::vector<int> params[][3] = {
- {IntArray(11), IntArray(-233), IntArray(0)},
- {IntArray(8), IntArray(-233), IntArray(0)},
- {IntArray(6), IntArray(-233), IntArray(1)},
- {IntArray(5), IntArray(-233), IntArray(2)},
- {IntArray(4), IntArray(-233), IntArray(-2)},
- {IntArray(6), IntArray(-233), IntArray(3)},
- {IntArray(5), IntArray(-233), IntArray(-1)},
- {IntArray(8, 4), IntArray(-233, -233), IntArray(0, 1)},
- {IntArray(12, 6), IntArray(-233, -233), IntArray(0, 2)},
- {IntArray(11, 5), IntArray(-233, -233), IntArray(-4, -2)},
- {IntArray(4, 4), IntArray(-233, -233), IntArray(1, 2)},
- {IntArray(12, 6), IntArray(-233, -233), IntArray(0, 3)},
- {IntArray(5, 5), IntArray(-233, -233), IntArray(1, 3)},
- {IntArray(4, 4), IntArray(-233, -233), IntArray(2, 3)},
- {IntArray(12, 6, 6), IntArray(-233, -233, -233), IntArray(0, 1, 2)},
- {IntArray(11, 5, 5), IntArray(-233, -233, -233), IntArray(0, 1, 2)},
- {IntArray(8, 4, 4), IntArray(-233, -233, -233), IntArray(0, 1, 3)},
- {IntArray(12, 6, 6), IntArray(-233, -233, -233), IntArray(0, 2, 3)},
- {IntArray(11, 5, 5), IntArray(-233, -233, -233), IntArray(0, 2, 3)},
- {IntArray(4, 4, 4), IntArray(-233, -233, -233), IntArray(1, 2, 3)},
- {IntArray(6, 6, 6), IntArray(-233, -233, -233), IntArray(1, 2, 3)},
- {IntArray(11, 5, 5, 5), IntArray(-233, -233, -233, -233), IntArray(0, 1, 2, 3)},
- {IntArray(8, 4, 4, 4), IntArray(-233, -233, -233, -233), IntArray(0, 1, 2, 3)},
- {IntArray(12, 6, 6, 6), IntArray(-233, -233, -233, -233), IntArray(-4, -3, -2, -1)},
- {IntArray(11), IntArray(11 + 16), IntArray(0)},
- {IntArray(12), IntArray(12 + 7), IntArray(0)},
- {IntArray(8), IntArray(8 + 12), IntArray(-4)},
- {IntArray(5), IntArray(11), IntArray(1)},
- {IntArray(6), IntArray(13), IntArray(1)},
- {IntArray(4), IntArray(12), IntArray(-3)},
- {IntArray(3), IntArray(12), IntArray(2)},
- {IntArray(4), IntArray(13), IntArray(2)},
- {IntArray(5), IntArray(11), IntArray(-2)},
- {IntArray(1), IntArray(8), IntArray(3)},
- {IntArray(2), IntArray(7), IntArray(3)},
- {IntArray(3), IntArray(6), IntArray(-1)},
- {IntArray(11, 5), IntArray(11 + 7, 11), IntArray(0, 1)},
- {IntArray(12, 6), IntArray(12 + 12, 12), IntArray(0, 1)},
- {IntArray(8, 4), IntArray(8 + 16, 13), IntArray(-4, -3)},
- {IntArray(11, 4), IntArray(11 + 12, 13), IntArray(0, 2)},
- {IntArray(12, 3), IntArray(12 + 16, 11), IntArray(0, 2)},
- {IntArray(8, 2), IntArray(8 + 7, 12), IntArray(-4, -2)},
- {IntArray(11, 1), IntArray(11 + 16, 5), IntArray(0, 3)},
- {IntArray(12, 2), IntArray(12 + 7, 6), IntArray(0, 3)},
- {IntArray(8, 3), IntArray(8 + 12, 7), IntArray(-4, -1)},
- {IntArray(3, 3), IntArray(13, 4), IntArray(1, 2)},
- {IntArray(4, 2), IntArray(12, 3), IntArray(1, 2)},
- {IntArray(5, 1), IntArray(11, 2), IntArray(-3, -2)},
- {IntArray(5, 5), IntArray(11, 8), IntArray(1, 3)},
- {IntArray(4, 6), IntArray(12, 9), IntArray(1, 3)},
- {IntArray(3, 4), IntArray(13, 7), IntArray(-3, -1)},
- {IntArray(2, 3), IntArray(12, 9), IntArray(2, 3)},
- {IntArray(3, 2), IntArray(11, 7), IntArray(2, 3)},
- {IntArray(4, 1), IntArray(10, 8), IntArray(-2, -1)},
- {IntArray(11, 2, 2), IntArray(11 + 6, 9, 9), IntArray(0, 1, 2)},
- {IntArray(12, 3, 3), IntArray(12 + 1, 10, 10), IntArray(0, 1, 2)},
- {IntArray(8, 4, 4), IntArray(8 + 3, 11, 11), IntArray(-4, -3, -2)},
- {IntArray(11, 4, 4), IntArray(11 + 12, 12, 12), IntArray(0, 1, 3)},
- {IntArray(12, 5, 5), IntArray(12 + 8, 11, 11), IntArray(0, 1, 3)},
- {IntArray(8, 6, 6), IntArray(8 + 4, 13, 13), IntArray(-4, -3, -1)},
- {IntArray(11, 1, 4), IntArray(11 + 5, 12, 12), IntArray(0, 2, 3)},
- {IntArray(12, 3, 3), IntArray(12 + 3, 11, 11), IntArray(0, 2, 3)},
- {IntArray(8, 2, 5), IntArray(8 + 2, 10, 10), IntArray(-4, -2, -1)},
- {IntArray(1, 1, 1), IntArray(7, 7, 7), IntArray(1, 2, 3)},
- {IntArray(2, 2, 2), IntArray(8, 9, 10), IntArray(1, 2, 3)},
- {IntArray(3, 3, 3), IntArray(11, 12, 13), IntArray(-3, -2, -1)},
- {IntArray(11, 2, 3, 6), IntArray(11 + 11, 10, 12, 11), IntArray(0, 1, 2, 3)},
- {IntArray(12, 3, 4, 5), IntArray(12 + 12, 9, 11, 13), IntArray(0, 1, 2, 3)},
- {IntArray(8, 4, 5, 4), IntArray(8 + 8, 8, 10, 12), IntArray(-4, -3, -2, -1)},
- {IntArray(11), IntArray(-7 + 1), IntArray(0)},
- {IntArray(12), IntArray(-12 + 1), IntArray(0)},
- {IntArray(8), IntArray(-16 + 1), IntArray(-4)},
- {IntArray(5), IntArray(-6 + 1), IntArray(1)},
- {IntArray(6), IntArray(-5 + 1), IntArray(1)},
- {IntArray(4), IntArray(-4 + 1), IntArray(-3)},
- {IntArray(4), IntArray(-4 + 1), IntArray(2)},
- {IntArray(5), IntArray(-5 + 1), IntArray(2)},
- {IntArray(6), IntArray(-6 + 1), IntArray(-2)},
- {IntArray(1), IntArray(-5 + 1), IntArray(3)},
- {IntArray(2), IntArray(-4 + 1), IntArray(3)},
- {IntArray(3), IntArray(-3 + 1), IntArray(-1)},
- {IntArray(11, 3), IntArray(-7 + 1, -3 + 1), IntArray(0, 1)},
- {IntArray(12, 4), IntArray(-12 + 1, -4 + 1), IntArray(0, 1)},
- {IntArray(8, 5), IntArray(-16 + 1, -5 + 1), IntArray(-4, -3)},
- {IntArray(11, 1), IntArray(-12 + 1, -5 + 1), IntArray(0, 2)},
- {IntArray(12, 2), IntArray(-16 + 1, -4 + 1), IntArray(0, 2)},
- {IntArray(8, 3), IntArray(-7 + 1, -6 + 1), IntArray(-4, -2)},
- {IntArray(11, 3), IntArray(-12 + 1, -2 + 1), IntArray(0, 3)},
- {IntArray(12, 4), IntArray(-16 + 1, -3 + 1), IntArray(0, 3)},
- {IntArray(8, 5), IntArray(-7 + 1, -4 + 1), IntArray(-4, -1)},
- {IntArray(2, 3), IntArray(-4 + 1, -2 + 1), IntArray(1, 2)},
- {IntArray(3, 4), IntArray(-2 + 1, -3 + 1), IntArray(1, 2)},
- {IntArray(4, 5), IntArray(-3 + 1, -4 + 1), IntArray(-3, -2)},
- {IntArray(3, 2), IntArray(-2 + 1, -4 + 1), IntArray(1, 3)},
- {IntArray(4, 3), IntArray(-3 + 1, -2 + 1), IntArray(1, 3)},
- {IntArray(5, 4), IntArray(-4 + 1, -3 + 1), IntArray(-3, -1)},
- {IntArray(2, 3), IntArray(-4 + 1, -6 + 1), IntArray(2, 3)},
- {IntArray(1, 2), IntArray(-5 + 1, -5 + 1), IntArray(2, 3)},
- {IntArray(3, 1), IntArray(-6 + 1, -4 + 1), IntArray(-2, -1)},
- {IntArray(11, 3, 3), IntArray(-7 + 1, -3 + 1, -4 + 1), IntArray(0, 1, 2)},
- {IntArray(12, 4, 4), IntArray(-12 + 1, -4 + 1, -3 + 1), IntArray(0, 1, 2)},
- {IntArray(8, 5, 5), IntArray(-16 + 1, -5 + 1, -5 + 1), IntArray(-4, -3, -2)},
- {IntArray(11, 2, 2), IntArray(-7 + 1, -5 + 1, -4 + 1), IntArray(0, 1, 3)},
- {IntArray(12, 1, 1), IntArray(-12 + 1, -6 + 1, -5 + 1), IntArray(0, 1, 3)},
- {IntArray(8, 3, 3), IntArray(-16 + 1, -4 + 1, -6 + 1), IntArray(-4, -3, -1)},
- {IntArray(11, 2, 5), IntArray(-7 + 1, -2 + 1, -5 + 1), IntArray(0, 2, 3)},
- {IntArray(12, 3, 3), IntArray(-12 + 1, -3 + 1, -4 + 1), IntArray(0, 2, 3)},
- {IntArray(8, 4, 4), IntArray(-16 + 1, -4 + 1, -3 + 1), IntArray(-4, -2, -1)},
- {IntArray(1, 3, 3), IntArray(-3 + 1, -6 + 1, -4 + 1), IntArray(1, 2, 3)},
- {IntArray(2, 2, 2), IntArray(-4 + 1, -4 + 1, -5 + 1), IntArray(1, 2, 3)},
- {IntArray(3, 1, 1), IntArray(-5 + 1, -5 + 1, -6 + 1), IntArray(-3, -2, -1)},
- {IntArray(11, 3, 4, 4), IntArray(-7 + 1, -3 + 1, -2 + 1, -4 + 1), IntArray(0, 1, 2, 3)},
- {IntArray(12, 4, 5, 3), IntArray(-12 + 1, -4 + 1, -3 + 1, -5 + 1), IntArray(0, 1, 2, 3)},
- {IntArray(8, 5, 6, 2), IntArray(-16 + 1, -5 + 1, -4 + 1, -3 + 1), IntArray(-4, -3, -2, -1)}
- };
-
- for (int i = 0; i < sizeof(params) / sizeof(params[0]); i++)
- {
- int ret = test_crop(a, params[i][0], params[i][1], params[i][2]);
- if (ret)
- return ret;
- }
-
- return 0;
- }
-
- int main()
- {
- SRAND(776757);
-
- return 0
- || test_crop_1d(RandomMat(112))
- || test_crop_1d(RandomMat(126))
- || test_crop_1d(RandomMat(127))
- || test_crop_2d(RandomMat(20, 48))
- || test_crop_2d(RandomMat(15, 36))
- || test_crop_2d(RandomMat(16, 33))
- || test_crop_3d(RandomMat(20, 20, 48))
- || test_crop_3d(RandomMat(15, 15, 36))
- || test_crop_3d(RandomMat(16, 16, 33))
- || test_crop_4d(RandomMat(20, 20, 20, 48))
- || test_crop_4d(RandomMat(15, 15, 15, 36))
- || test_crop_4d(RandomMat(16, 16, 16, 33));
- }
|