Files
ncnn/tests/test_crop_1.cpp
2026-10-01 23:33:05 +08:00

378 lines
17 KiB
C++

// 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, int flag = 0)
{
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, 0.001, flag);
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, int flag = 0)
{
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], flag);
if (ret)
return ret;
}
return 0;
}
static int test_crop_2d(const ncnn::Mat& a, int flag = 0)
{
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], flag);
if (ret)
return ret;
}
return 0;
}
static int test_crop_3d(const ncnn::Mat& a, int flag = 0)
{
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], flag);
if (ret)
return ret;
}
return 0;
}
static int test_crop_4d(const ncnn::Mat& a, int flag = 0)
{
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], flag);
if (ret)
return ret;
}
return 0;
}
// the 36-channel cases retain Vulkan pack4 crops; keep all 1d packing cases
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_LAYER_DISABLE_GPU_TESTING)
|| test_crop_2d(RandomMat(15, 36))
|| test_crop_2d(RandomMat(16, 33))
|| test_crop_3d(RandomMat(20, 20, 48), TEST_LAYER_DISABLE_GPU_TESTING)
|| test_crop_3d(RandomMat(15, 15, 36))
|| test_crop_3d(RandomMat(16, 16, 33))
|| test_crop_4d(RandomMat(20, 20, 20, 48), TEST_LAYER_DISABLE_GPU_TESTING)
|| test_crop_4d(RandomMat(15, 15, 15, 36))
|| test_crop_4d(RandomMat(16, 16, 16, 33));
}