GithubStar/ncnn
1
0
Fork 0
mirror of https://github.com/Tencent/ncnn.git synced 2026-10-18 13:42:18 +08:00
Code Issues Packages Projects Releases Wiki Activity
Files
master
ncnn/tests/test_crop_oom.cpp
T

184 lines
6.9 KiB
C++
Raw Permalink Blame History

// Copyright 2025 Tencent
// SPDX-License-Identifier: BSD-3-Clause
#include "testutil.h"
static int test_crop_oom(const ncnn::Mat& a, int woffset, int hoffset, int doffset, int coffset, int outw, int outh, int outd, int outc, int woffset2, int hoffset2, int doffset2, int coffset2)
{
ncnn::ParamDict pd;
pd.set(0, woffset);
pd.set(1, hoffset);
pd.set(13, doffset);
pd.set(2, coffset);
pd.set(3, outw);
pd.set(4, outh);
pd.set(14, outd);
pd.set(5, outc);
pd.set(6, woffset2);
pd.set(7, hoffset2);
pd.set(15, doffset2);
pd.set(8, coffset2);
std::vector<ncnn::Mat> weights(0);
int ret = test_layer_oom("Crop", pd, weights, a);
if (ret != 0)
{
fprintf(stderr, "test_crop_oom failed a.dims=%d a=(%d %d %d %d) woffset=%d hoffset=%d doffset=%d coffset=%d outw=%d outh=%d outd=%d outc=%d woffset2=%d hoffset2=%d doffset2=%d coffset2=%d\n", a.dims, a.w, a.h, a.d, a.c, woffset, hoffset, doffset, coffset, outw, outh, outd, outc, woffset2, hoffset2, doffset2, coffset2);
}
return ret;
}
static int test_crop_oom(const ncnn::Mat& a, const char* starts_expr, const char* ends_expr, const char* axes_expr)
{
ncnn::ParamDict pd;
pd.set(19, std::string(starts_expr));
pd.set(20, std::string(ends_expr));
pd.set(21, std::string(axes_expr));
std::vector<ncnn::Mat> weights(0);
int ret = test_layer_oom("Crop", pd, weights, a);
if (ret != 0)
{
fprintf(stderr, "test_crop_oom failed a.dims=%d a=(%d %d %d %d) starts_expr=%s ends_expr=%s axes_expr=%s\n", a.dims, a.w, a.h, a.d, a.c, starts_expr, ends_expr, axes_expr);
}
return ret;
}
static int test_crop_oom(const std::vector<ncnn::Mat>& as, const char* starts_expr, const char* ends_expr, const char* axes_expr)
{
ncnn::ParamDict pd;
pd.set(19, std::string(starts_expr));
pd.set(20, std::string(ends_expr));
pd.set(21, std::string(axes_expr));
std::vector<ncnn::Mat> weights(0);
int ret = test_layer_oom("Crop", pd, weights, as, 1);
if (ret != 0)
{
fprintf(stderr, "test_crop_oom failed a.dims=%d a=(%d %d %d %d) starts_expr=%s ends_expr=%s axes_expr=%s\n", as[0].dims, as[0].w, as[0].h, as[0].d, as[0].c, starts_expr, ends_expr, axes_expr);
}
return ret;
}
static int test_crop_0()
{
ncnn::Mat a = RandomMat(13, 12, 25, 48);
ncnn::Mat b = RandomMat(13, 12, 48);
ncnn::Mat c = RandomMat(13, 48);
ncnn::Mat d = RandomMat(128);
return 0
|| test_crop_oom(a, 1, 1, 1, 1, -233, -233, -233, -233, 1, 1, 1, 1)
|| test_crop_oom(b, 1, 1, 0, 1, -233, -233, 0, -233, 1, 1, 0, 1)
|| test_crop_oom(c, 1, 1, 0, 0, -233, -233, 0, 0, 1, 1, 0, 0)
|| test_crop_oom(d, 1, 0, 0, 0, -233, 0, 0, 0, 1, 0, 0, 0)
|| test_crop_oom(a, 2, 2, 2, 2, 6, 6, 6, 16, 0, 0, 0, 0)
|| test_crop_oom(b, 2, 2, 0, 2, 6, 6, 0, 16, 0, 0, 0, 0)
|| test_crop_oom(c, 2, 2, 0, 0, 6, 16, 0, 0, 0, 0, 0, 0)
|| test_crop_oom(d, 2, 0, 0, 0, 16, 0, 0, 0, 0, 0, 0, 0)
|| test_crop_oom(a, 3, 3, 3, 16, 3, 4, 5, 16, 0, 0, 0, 0)
|| test_crop_oom(b, 3, 3, 0, 16, 3, 4, 0, 16, 0, 0, 0, 0)
|| test_crop_oom(c, 3, 16, 0, 0, 3, 16, 0, 0, 0, 0, 0, 0)
|| test_crop_oom(d, 16, 0, 0, 0, 32, 0, 0, 0, 0, 0, 0, 0);
}
static int test_crop_1()
{
ncnn::Mat a = RandomMat(13, 12, 25, 47);
ncnn::Mat b = RandomMat(13, 12, 47);
ncnn::Mat c = RandomMat(13, 47);
ncnn::Mat d = RandomMat(129);
return 0
|| test_crop_oom(a, 1, 1, 1, 1, -233, -233, -233, -233, 1, 1, 1, 1)
|| test_crop_oom(b, 1, 1, 0, 1, -233, -233, 0, -233, 1, 1, 0, 1)
|| test_crop_oom(c, 1, 1, 0, 0, -233, -233, 0, 0, 1, 1, 0, 0)
|| test_crop_oom(d, 1, 0, 0, 0, -233, 0, 0, 0, 1, 0, 0, 0)
|| test_crop_oom(a, 2, 2, 2, 2, 6, 6, 6, 16, 0, 0, 0, 0)
|| test_crop_oom(b, 2, 2, 0, 2, 6, 6, 0, 16, 0, 0, 0, 0)
|| test_crop_oom(c, 2, 2, 0, 0, 6, 16, 0, 0, 0, 0, 0, 0)
|| test_crop_oom(d, 2, 0, 0, 0, 16, 0, 0, 0, 0, 0, 0, 0)
|| test_crop_oom(a, 3, 3, 3, 16, 6, 6, 6, 16, 0, 0, 0, 0)
|| test_crop_oom(b, 3, 3, 0, 16, 6, 6, 0, 16, 0, 0, 0, 0)
|| test_crop_oom(c, 3, 16, 0, 0, 6, 16, 0, 0, 0, 0, 0, 0)
|| test_crop_oom(d, 16, 0, 0, 0, 32, 0, 0, 0, 0, 0, 0, 0);
}
static int test_crop_2()
{
ncnn::Mat a = RandomMat(13, 12, 25, 48);
ncnn::Mat b = RandomMat(13, 12, 48);
ncnn::Mat c = RandomMat(13, 48);
ncnn::Mat d = RandomMat(128);
return 0
|| test_crop_oom(a, "2", "-2", "0")
|| test_crop_oom(b, "2", "-2", "0")
|| test_crop_oom(c, "2", "-2", "0")
|| test_crop_oom(d, "2", "-2", "0")
|| test_crop_oom(a, "16", "32", "-4")
|| test_crop_oom(b, "16", "32", "-3")
|| test_crop_oom(c, "16", "32", "-2")
|| test_crop_oom(d, "16", "32", "-1")
|| test_crop_oom(a, "16,//(0d,4),2,1", "32,-1,-2,-3", "0,1,2,3")
|| test_crop_oom(b, "16,//(0h,4),2", "32,-1,-(0w,2)", "0,1,2")
|| test_crop_oom(c, "16,//(0w,4)", "32,-2", "0,1")
|| test_crop_oom(a, "10", "11", "1")
|| test_crop_oom(b, "1,1", "-(0c,15),-(0w,5)", "0,2")
|| test_crop_oom(a, "-(0w,3),0h//2,floor(*(0c,0.3))", "-1,0h,ceil(*(0c,0.9))", "3,2,0")
|| test_crop_oom(b, "-(0w,3),0h//2,floor(*(0c,0.3))", "-1,0h,ceil(*(0c,0.9))", "2,1,0")
|| test_crop_oom(c, "-(0w,3),floor(*(0h,0.3))", "-1,ceil(*(0h,0.9))", "1,0")
|| test_crop_oom(d, "floor(*(0w,0.3))", "ceil(*(0w,0.9))", "0");
}
static int test_crop_3()
{
std::vector<ncnn::Mat> as(2);
as[0] = RandomMat(14, 15, 13, 48);
as[1] = RandomMat(8, 5, 3, 4);
std::vector<ncnn::Mat> bs(2);
bs[0] = RandomMat(14, 15, 48);
bs[1] = RandomMat(28, 45, 16);
std::vector<ncnn::Mat> cs(2);
cs[0] = RandomMat(24, 48);
cs[1] = RandomMat(28, 6);
std::vector<ncnn::Mat> ds(3);
ds[0] = RandomMat(128);
ds[1] = RandomMat(16);
ds[2] = RandomMat(64);
return 0
|| test_crop_oom(as, "*(1c,4)", "*(1c,8)", "-4")
|| test_crop_oom(bs, "1c", "-(0c,1c)", "-3")
|| test_crop_oom(cs, "+(1h,10)", "-(1h,22)", "-2")
|| test_crop_oom(ds, "1w", "2w", "-1")
|| test_crop_oom(as, "16,//(min(0w,1d),4),2,3", "32,-1,-2,-3", "0,1,2,3")
|| test_crop_oom(bs, "16,//(min(0w,1h),4),2", "32,-1,-(0w,2)", "0,1,2")
|| test_crop_oom(cs, "16,//(min(0w,1w),4)", "32,-2", "0,1")
|| test_crop_oom(bs, "1,//(1w,7)", "+(1c,1),-(0w,2)", "0,2")
|| test_crop_oom(as, "-(1w,4)", "neg(1h,3)", "0")
|| test_crop_oom(bs, "-(1w,20)", "-2", "0")
|| test_crop_oom(bs, "//(1h,15)", "neg(//(1w,7))", "2")
|| test_crop_oom(bs, "//(100,0h),round(fmod(100,0c))", "-233,min(1c,0c)", "1,0");
}
int main()
{
SRAND(776757);
return 0
|| test_crop_0()
|| test_crop_1()
|| test_crop_2()
|| test_crop_3();
}
Reference in New Issue View Git Blame Copy Permalink