potok/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2026-07-31 00:03:03 +04:00
Code Packages Releases Wiki Activity

add DetectionOutputOp

Browse Source
This commit is contained in:
YashasSamaga
2020-07-01 23:47:57 +05:30
parent afe9993376
commit ae293f27cf
16 changed files with 1985 additions and 73 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/dnn/test/test_tf_importer.cpp
+9 -2
View File
@@ -691,6 +691,13 @@ TEST_P(Test_TensorFlow_nets, Faster_RCNN)
checkBackend();
double scoresDiff = backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 ? 2.9e-5 : 1e-5;
double iouDiff = 1e-4;
if (target == DNN_TARGET_CUDA)
{
// for faster_rcnn_resnet50_coco_2018_01_28
scoresDiff = 0.06;
iouDiff = 0.08;
}
for (int i = 0; i < 2; ++i)
{
std::string proto = findDataFile("dnn/" + names[i] + ".pbtxt");
@@ -706,7 +713,7 @@ TEST_P(Test_TensorFlow_nets, Faster_RCNN)
Mat out = net.forward();
Mat ref = blobFromNPY(findDataFile("dnn/tensorflow/" + names[i] + ".detection_out.npy"));
normAssertDetections(ref, out, names[i].c_str(), 0.3, scoresDiff);
normAssertDetections(ref, out, names[i].c_str(), 0.3, scoresDiff, iouDiff);
}
}
@@ -1244,7 +1251,7 @@ TEST_P(Test_TensorFlow_nets, EfficientDet)
if (target == DNN_TARGET_CUDA_FP16)
{
scoreDiff = 0.002;
iouDiff = 0.003;
iouDiff = 0.004;
}
normAssertDetections(ref, out, "", 0.5, scoreDiff, iouDiff);
expectNoFallbacksFromIE(net);