Skip to content

Commit 21f0146

Browse files
committed
Merge pull request opencv#14212 from dkurt:reduce_ie_models_tests
2 parents 4516957 + 144b1f4 commit 21f0146

File tree

1 file changed

+15
-60
lines changed

1 file changed

+15
-60
lines changed

modules/dnn/test/test_ie_models.cpp

Lines changed: 15 additions & 60 deletions
Original file line numberDiff line numberDiff line change
@@ -177,34 +177,17 @@ TEST_P(DNNTestOpenVINO, models)
177177
{
178178
Target target = (dnn::Target)(int)get<0>(GetParam());
179179
std::string modelName = get<1>(GetParam());
180+
std::string precision = (target == DNN_TARGET_OPENCL_FP16 || target == DNN_TARGET_MYRIAD) ? "FP16" : "FP32";
180181

181182
#ifdef INF_ENGINE_RELEASE
182-
#if INF_ENGINE_RELEASE <= 2018030000
183-
if (target == DNN_TARGET_MYRIAD && (modelName == "landmarks-regression-retail-0001" ||
184-
modelName == "semantic-segmentation-adas-0001" ||
185-
modelName == "face-reidentification-retail-0001"))
186-
throw SkipTestException("");
187-
#elif INF_ENGINE_RELEASE == 2018040000
188-
if (modelName == "single-image-super-resolution-0034" ||
189-
(target == DNN_TARGET_MYRIAD && (modelName == "license-plate-recognition-barrier-0001" ||
190-
modelName == "landmarks-regression-retail-0009" ||
191-
modelName == "semantic-segmentation-adas-0001")))
192-
throw SkipTestException("");
193-
#elif INF_ENGINE_RELEASE == 2018050000
194-
if (modelName == "single-image-super-resolution-0063" ||
195-
modelName == "single-image-super-resolution-1011" ||
196-
modelName == "single-image-super-resolution-1021" ||
197-
(target == DNN_TARGET_OPENCL_FP16 && modelName == "face-reidentification-retail-0095") ||
198-
(target == DNN_TARGET_MYRIAD && (modelName == "license-plate-recognition-barrier-0001" ||
199-
modelName == "semantic-segmentation-adas-0001")))
200-
throw SkipTestException("");
201-
#endif
202-
#endif
203-
204-
std::string precision = (target == DNN_TARGET_OPENCL_FP16 || target == DNN_TARGET_MYRIAD) ? "FP16" : "FP32";
183+
#if INF_ENGINE_RELEASE <= 2018050000
205184
std::string prefix = utils::fs::join("intel_models",
206185
utils::fs::join(modelName,
207186
utils::fs::join(precision, modelName)));
187+
#endif
188+
#endif
189+
190+
initDLDTDataPath();
208191
std::string xmlPath = findDataFile(prefix + ".xml");
209192
std::string binPath = findDataFile(prefix + ".bin");
210193

@@ -221,49 +204,21 @@ TEST_P(DNNTestOpenVINO, models)
221204
{
222205
auto dstIt = cvOutputsMap.find(srcIt.first);
223206
CV_Assert(dstIt != cvOutputsMap.end());
224-
double normInfIE = cvtest::norm(srcIt.second, cv::NORM_INF);
225207
double normInf = cvtest::norm(srcIt.second, dstIt->second, cv::NORM_INF);
226-
double eps = 0;
227-
if (target == DNN_TARGET_OPENCL_FP16 || target == DNN_TARGET_MYRIAD)
228-
{
229-
double fp16_eps = 1.0/1024;
230-
eps = fp16_eps * 1/*ULP*/ * std::max(normInfIE, 1.0);
231-
}
232-
EXPECT_LE(normInf, eps) << "IE: " << normInfIE;
233-
}
234-
}
235-
236-
static testing::internal::ParamGenerator<String> intelModels()
237-
{
238-
initDLDTDataPath();
239-
std::vector<String> modelsNames;
240-
241-
std::string path;
242-
try
243-
{
244-
path = findDataDirectory("intel_models", false);
245-
}
246-
catch (...)
247-
{
248-
std::cerr << "ERROR: Can't find OpenVINO models. Check INTEL_CVSDK_DIR environment variable (run setup.sh)" << std::endl;
249-
return ValuesIn(modelsNames); // empty list
208+
EXPECT_EQ(normInf, 0);
250209
}
251-
252-
cv::utils::fs::glob_relative(path, "", modelsNames, false, true);
253-
254-
modelsNames.erase(
255-
std::remove_if(modelsNames.begin(), modelsNames.end(),
256-
[&](const String& dir){ return !utils::fs::isDirectory(utils::fs::join(path, dir)); }),
257-
modelsNames.end()
258-
);
259-
CV_Assert(!modelsNames.empty());
260-
261-
return ValuesIn(modelsNames);
262210
}
263211

264212
INSTANTIATE_TEST_CASE_P(/**/,
265213
DNNTestOpenVINO,
266-
Combine(testing::ValuesIn(getAvailableTargets(DNN_BACKEND_INFERENCE_ENGINE)), intelModels())
214+
Combine(testing::ValuesIn(getAvailableTargets(DNN_BACKEND_INFERENCE_ENGINE)),
215+
testing::Values(
216+
"age-gender-recognition-retail-0013",
217+
"face-person-detection-retail-0002",
218+
"head-pose-estimation-adas-0001",
219+
"person-detection-retail-0002",
220+
"vehicle-detection-adas-0002"
221+
))
267222
);
268223

269224
}}

0 commit comments

Comments
 (0)