mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-08-05 22:19:14 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

dnn(test): fix outputs handling in ONNX conformance

Browse Source 
- ONNX output is 1 tensor per defined output instead of N tensors from outputs of "output" layer
This commit is contained in:
Alexander Alekhin 2022-01-29 19:58:46 +00:00
parent 85719a0a5d
commit a7e6a1059c
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
modules/dnn/test/test_onnx_conformance.cpp
View File

@ -1171,10 +1171,10 @@ TEST_P(Test_ONNX_conformance, Layer_Test)
}
std::vector<String> layerNames = net.getUnconnectedOutLayersNames();
std::vector< std::vector<Mat> > outputs_;
std::vector<Mat> outputs;
try
{
net.forward(outputs_, layerNames);
net.forward(outputs, layerNames);
}
catch (...)
{
@ -1182,8 +1182,7 @@ TEST_P(Test_ONNX_conformance, Layer_Test)
applyTestTag(CV_TEST_TAG_DNN_ERROR_FORWARD);
throw;
}
ASSERT_GE(outputs_.size(), 1);
const std::vector<Mat>& outputs = outputs_[0];
ASSERT_GE(outputs.size(), 1);
if (checkLayersFallbacks && checkFallbacks(net))
{