IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

NMS: change back threshold for comparison (#8553)

Browse Source
This commit is contained in:
Zhibin Li 2021-11-16 12:49:50 +08:00 committed by GitHub
parent 9676e643cf
commit 4edb7c6f34
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
inference-engine/tests/functional/shared_test_classes/src/single_layer/non_max_suppression.cpp
View File

@ -91,12 +91,12 @@ void NmsLayerTest::CompareBuffer(const std::vector<std::pair<ngraph::element::Ty
case ngraph::element::Type_t::f32:
LayerTestsUtils::LayerTestsCommon::Compare(
reinterpret_cast<const float *>(expectedBuffer),
reinterpret_cast<const float *>(actualBuffer), size, 0);
reinterpret_cast<const float *>(actualBuffer), size, threshold);
break;
case ngraph::element::Type_t::f64:
LayerTestsUtils::LayerTestsCommon::Compare(
reinterpret_cast<const double *>(expectedBuffer),
reinterpret_cast<const float *>(actualBuffer), size, 0);
reinterpret_cast<const float *>(actualBuffer), size, threshold);
break;
default:
break;