mirror of
https://github.com/Tencent/ncnn.git
synced 2026-10-19 06:21:39 +08:00
nihui
111 lines
3.7 KiB
C++
111 lines
3.7 KiB
C++
// Copyright 2025 Tencent
|
|
// SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
#include "testutil.h"
|
|
|
|
static int test_crop(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("Crop", pd, weights, a);
|
|
if (ret != 0)
|
|
{
|
|
fprintf(stderr, "test_crop 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(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("Crop", pd, weights, as, 1);
|
|
if (ret != 0)
|
|
{
|
|
fprintf(stderr, "test_crop 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(a, "2", "-2", "0")
|
|
|| test_crop(b, "2", "-2", "0")
|
|
|| test_crop(c, "2", "-2", "0")
|
|
|| test_crop(d, "2", "-2", "0")
|
|
|| test_crop(a, "16", "32", "-4")
|
|
|| test_crop(b, "16", "32", "-3")
|
|
|| test_crop(c, "16", "32", "-2")
|
|
|| test_crop(d, "16", "32", "-1")
|
|
|| test_crop(a, "16,//(0d,4),2,3", "32,-1,-2,-3", "0,1,2,3")
|
|
|| test_crop(b, "16,//(0h,4),2", "32,-1,-(0w,2)", "0,1,2")
|
|
|| test_crop(c, "16,//(0w,4)", "32,-2", "0,1")
|
|
|| test_crop(a, "10", "11", "1")
|
|
|| test_crop(b, "1,1", "-(0c,15),-(0w,5)", "0,2")
|
|
|| test_crop(a, "-(0w,3),0h//2,floor(*(0c,0.3))", "-1,0h,ceil(*(0c,0.9))", "3,2,0")
|
|
|| test_crop(b, "-(0w,3),0h//2,floor(*(0c,0.3))", "-1,0h,ceil(*(0c,0.9))", "2,1,0")
|
|
|| test_crop(c, "-(0w,3),floor(*(0h,0.3))", "-1,ceil(*(0h,0.9))", "1,0")
|
|
|| test_crop(d, "floor(*(0w,0.3))", "ceil(*(0w,0.9))", "0");
|
|
}
|
|
|
|
static int test_crop_1()
|
|
{
|
|
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(as, "*(1c,4)", "*(1c,8)", "-4")
|
|
|| test_crop(bs, "1c", "-(0c,1c)", "-3")
|
|
|| test_crop(cs, "+(1h,10)", "-(1h,22)", "-2")
|
|
|| test_crop(ds, "1w", "2w", "-1")
|
|
|| test_crop(as, "16,//(min(0w,1d),4),2,3", "32,-1,-2,-3", "0,1,2,3")
|
|
|| test_crop(bs, "16,//(min(0w,1h),4),2", "32,-1,-(0w,2)", "0,1,2")
|
|
|| test_crop(cs, "16,//(min(0w,1w),4)", "32,-2", "0,1")
|
|
|| test_crop(bs, "1,//(1w,7)", "+(1c,1),-(0w,2)", "0,2")
|
|
|| test_crop(as, "-(1w,4)", "neg(1h,3)", "0")
|
|
|| test_crop(bs, "-(1w,20)", "-2", "0")
|
|
|| test_crop(bs, "//(1h,15)", "neg(//(1w,7))", "2")
|
|
|| test_crop(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();
|
|
}
|