Fixed Test_Model.DetectionOutput

This commit is contained in:
Liubov Batanina 2020-11-19 12:11:52 +03:00
parent e8536c4a0e
commit b86f129393
2 changed files with 3 additions and 1 deletions

View File

@ -100,7 +100,7 @@ public:
// Faster-RCNN or R-FCN
if (net.getLayer(0)->outputNameToIndex("im_info") != -1)
{
Mat imInfo(Matx31f(size.height, size.width, 1.6f));
Mat imInfo(Matx13f(size.height, size.width, 1.6f));
net.setInput(imInfo, "im_info");
}
net.forward(outs, outNames);

View File

@ -206,6 +206,8 @@ TEST_P(Test_Model, DetectionOutput)
{
if (backend == DNN_BACKEND_OPENCV)
scoreDiff = 4e-3;
else
scoreDiff = 2e-2;
iouDiff = 1.8e-1;
}