58 Star 280 Fork 3

腾讯开源/ncnn

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
test_interp_1.cpp 3.42 KB
一键复制 编辑 原始数据 按行查看 历史
// Tencent is pleased to support the open source community by making ncnn available.
//
// Copyright (C) 2025 THL A29 Limited, a Tencent company. All rights reserved.
//
// Licensed under the BSD 3-Clause License (the "License"); you may not use this file except
// in compliance with the License. You may obtain a copy of the License at
//
// https://opensource.org/licenses/BSD-3-Clause
//
// Unless required by applicable law or agreed to in writing, software distributed
// under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
// CONDITIONS OF ANY KIND, either express or implied. See the License for the
// specific language governing permissions and limitations under the License.
#include "testutil.h"
static int test_interp(const ncnn::Mat& a, int resize_type, int align_corner, const char* shape_expr)
{
ncnn::ParamDict pd;
pd.set(0, resize_type);
pd.set(5, 1); // dynamic_target_size
pd.set(6, align_corner);
pd.set(9, std::string(shape_expr));
std::vector<ncnn::Mat> weights(0);
int ret = test_layer("Interp", pd, weights, a);
if (ret != 0)
{
fprintf(stderr, "test_interp failed a.dims=%d a=(%d %d %d %d) resize_type=%d align_corner=%d shape_expr=%s\n", a.dims, a.w, a.h, a.d, a.c, resize_type, align_corner, shape_expr);
}
return ret;
}
static int test_interp(const std::vector<ncnn::Mat>& as, int resize_type, int align_corner, const char* shape_expr)
{
ncnn::ParamDict pd;
pd.set(0, resize_type);
pd.set(5, 1); // dynamic_target_size
pd.set(6, align_corner);
pd.set(9, std::string(shape_expr));
std::vector<ncnn::Mat> weights(0);
int ret = test_layer("Interp", pd, weights, as, 1);
if (ret != 0)
{
fprintf(stderr, "test_interp failed a.dims=%d a=(%d %d %d %d) resize_type=%d align_corner=%d shape_expr=%s\n", as[0].dims, as[0].w, as[0].h, as[0].d, as[0].c, resize_type, align_corner, shape_expr);
}
return ret;
}
static int test_interp_0()
{
ncnn::Mat a = RandomMat(32, 25, 3);
ncnn::Mat b = RandomMat(20, 14, 32);
return 0
|| test_interp(a, 1, 0, "0w,0h")
|| test_interp(a, 2, 0, "0w,5")
|| test_interp(b, 1, 0, "0w,0h")
|| test_interp(b, 2, 0, "0w,5")
|| test_interp(a, 3, 0, "square(neg(abs(max(min(//(*(-(+(0w,1),-1),4),4),100),-4)))),0h")
|| test_interp(a, 1, 0, "square(neg(abs(max(min(//(*(-(+(0w,1.0),-1.0),2.0),2.0),100.3),-2.2)))),0h")
|| test_interp(a, 2, 1, "0w,trunc(*(round(-(floor(+(ceil(*(0w,1.2)),0.7)),-0.4)),1.0001))")
|| test_interp(a, 3, 0, "*(0d,4),ceil(sqrt(square(*(asinh(sinh(atanh(tanh(atan(tan(acos(cos(asin(sin(/(0w,2))))))))))),16))))");
}
static int test_interp_1()
{
std::vector<ncnn::Mat> as(2);
as[0] = RandomMat(14, 15, 16);
as[1] = RandomMat(28, 45, 48);
return 0
|| test_interp(as, 1, 0, "*(1w,0.5),/(1h,3)")
|| test_interp(as, 2, 0, "*(0w,0h),-(-(1c,0c),16)");
}
static int test_interp_2()
{
ncnn::Mat a = RandomMat(14, 15, 16);
ncnn::Mat b = RandomMat(22, 13);
return 0
|| test_interp(a, 1, 0, "*(0w,0.5),/(0h,3)")
|| test_interp(a, 1, 0, "1,4")
|| test_interp(a, 2, 1, "*(0w,0h),0c")
|| test_interp(b, 1, 0, "//(0w,3)")
|| test_interp(b, 2, 0, "-(0w,5)");
}
int main()
{
SRAND(7767517);
return 0
|| test_interp_0()
|| test_interp_1()
|| test_interp_2();
}
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
C/C++
1
https://gitee.com/Tencent/ncnn.git
git@gitee.com:Tencent/ncnn.git
Tencent
ncnn
ncnn
master

搜索帮助