mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-06-13 04:52:53 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

dnn(test): adjust EAST test tolerance

Browse Source
This commit is contained in:
Alexander Alekhin 2020-12-10 15:51:58 +00:00
parent 114848d313
commit 8ff27a07bb
1 changed files with 27 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
modules/dnn/test/test_model.cpp
View File

@ -209,14 +209,15 @@ public:
normAssertTextDetections(gt, contours, "", 0.05f); normAssertTextDetections(gt, contours, "", 0.05f);
} }
void testTextDetectionModelByEAST(const std::string& weights, const std::string& cfg, void testTextDetectionModelByEAST(
const std::string& weights, const std::string& cfg,
const std::string& imgPath, const std::vector<RotatedRect>& gt, const std::string& imgPath, const std::vector<RotatedRect>& gt,
float confThresh, float nmsThresh, float confThresh, float nmsThresh,
const Size& size = {-1, -1}, Scalar mean = Scalar(), const Size& size = {-1, -1}, Scalar mean = Scalar(),
double scale = 1.0, bool swapRB = false, bool crop = false) double scale = 1.0, bool swapRB = false, bool crop = false,
double eps_center = 5/*pixels*/, double eps_size = 5/*pixels*/, double eps_angle = 1
)
{ {
const double EPS_PIXELS = 3;
checkBackend(); checkBackend();
Mat frame = imread(imgPath); Mat frame = imread(imgPath);
@ -255,11 +256,11 @@ public:
waitKey(0); waitKey(0);
#endif #endif
const RotatedRect& gtBox = gt[i]; const RotatedRect& gtBox = gt[i];
EXPECT_NEAR(box.center.x, gtBox.center.x, EPS_PIXELS); EXPECT_NEAR(box.center.x, gtBox.center.x, eps_center);
EXPECT_NEAR(box.center.y, gtBox.center.y, EPS_PIXELS); EXPECT_NEAR(box.center.y, gtBox.center.y, eps_center);
EXPECT_NEAR(box.size.width, gtBox.size.width, EPS_PIXELS); EXPECT_NEAR(box.size.width, gtBox.size.width, eps_size);
EXPECT_NEAR(box.size.height, gtBox.size.height, EPS_PIXELS); EXPECT_NEAR(box.size.height, gtBox.size.height, eps_size);
EXPECT_NEAR(box.angle, gtBox.angle, 1); EXPECT_NEAR(box.angle, gtBox.angle, eps_angle);
} }
} }
}; };
@ -642,9 +643,6 @@ TEST_P(Test_Model, TextDetectionByDB)
TEST_P(Test_Model, TextDetectionByEAST) TEST_P(Test_Model, TextDetectionByEAST)
{ {
if (target == DNN_TARGET_OPENCL_FP16)
applyTestTag(CV_TEST_TAG_DNN_SKIP_OPENCL_FP16);
std::string imgPath = _tf("text_det_test2.jpg"); std::string imgPath = _tf("text_det_test2.jpg");
std::string weightPath = _tf("frozen_east_text_detection.pb", false); std::string weightPath = _tf("frozen_east_text_detection.pb", false);
@ -663,7 +661,20 @@ TEST_P(Test_Model, TextDetectionByEAST)
float confThresh = 0.5; float confThresh = 0.5;
float nmsThresh = 0.4; float nmsThresh = 0.4;
testTextDetectionModelByEAST(weightPath, "", imgPath, gt, confThresh, nmsThresh, size, mean, scale, swapRB); double eps_center = 5/*pixels*/;
double eps_size = 5/*pixels*/;
double eps_angle = 1;
if (target == DNN_TARGET_OPENCL_FP16 || target == DNN_TARGET_CUDA_FP16 || target == DNN_TARGET_MYRIAD)
{
eps_center = 10;
eps_size = 25;
eps_angle = 3;
}
testTextDetectionModelByEAST(weightPath, "", imgPath, gt, confThresh, nmsThresh, size, mean, scale, swapRB, false/*crop*/,
eps_center, eps_size, eps_angle
);
} }
INSTANTIATE_TEST_CASE_P(/**/, Test_Model, dnnBackendsAndTargets()); INSTANTIATE_TEST_CASE_P(/**/, Test_Model, dnnBackendsAndTargets());