mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2024-11-28 13:10:12 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #24737 from opencv-pushbot:gitee/alalek/issue_24735

Browse Source 
dnn(test): tune FP16 test tolerance
This commit is contained in:
Alexander Smorkalov 2023-12-21 17:58:33 +03:00 committed by GitHub
commit 011fd39091
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
modules/dnn/test/test_onnx_importer.cpp
View File

@ -2959,12 +2959,6 @@ TEST_P(Test_ONNX_layers, AttentionSingleHead) {
TEST_P(Test_ONNX_nets, ViT_B_32) {
applyTestTag(CV_TEST_TAG_LONG, CV_TEST_TAG_DEBUG_LONG);
if (backend == DNN_BACKEND_CUDA && target == DNN_TARGET_CUDA_FP16)
{
// does not pass test for now
applyTestTag(CV_TEST_TAG_DNN_SKIP_CUDA_FP16);
}
const std::string model_path = _tf("models/vit_b_32.onnx", false);
auto net = readNet(model_path);
@ -2981,7 +2975,20 @@ TEST_P(Test_ONNX_nets, ViT_B_32) {
net.setInput(blob);
auto out = net.forward();
normAssert(ref, out, "ViTB_32", default_l1, default_lInf);
double l1 = default_l1;
double lInf = default_lInf;
if (target == DNN_TARGET_CUDA_FP16)
{
l1 = 0.01;
lInf = 0.05;
}
if (target == DNN_TARGET_OPENCL_FP16)
{
l1 = 0.008;
lInf = 0.04;
}
normAssert(ref, out, "ViTB_32", l1, lInf);
}
TEST_P(Test_ONNX_nets, VitTrack) {