Commit 80d1f2dd authored by Li Peng's avatar Li Peng

re-enable tensor flow mobilenet ssd ocl test

Signed-off-by: 's avatarLi Peng <peng.li@intel.com>
parent 5992c466
...@@ -174,9 +174,7 @@ TEST_P(DNNTestNetwork, MobileNet_SSD_Caffe) ...@@ -174,9 +174,7 @@ TEST_P(DNNTestNetwork, MobileNet_SSD_Caffe)
TEST_P(DNNTestNetwork, MobileNet_SSD_TensorFlow) TEST_P(DNNTestNetwork, MobileNet_SSD_TensorFlow)
{ {
if (backend == DNN_BACKEND_DEFAULT && target == DNN_TARGET_OPENCL || if (backend == DNN_BACKEND_HALIDE) throw SkipTestException("");
backend == DNN_BACKEND_HALIDE)
throw SkipTestException("");
Mat sample = imread(findDataFile("dnn/street.png", false)); Mat sample = imread(findDataFile("dnn/street.png", false));
Mat inp = blobFromImage(sample, 1.0f / 127.5, Size(300, 300), Scalar(127.5, 127.5, 127.5), false); Mat inp = blobFromImage(sample, 1.0f / 127.5, Size(300, 300), Scalar(127.5, 127.5, 127.5), false);
processNet("dnn/ssd_mobilenet_v1_coco.pb", "dnn/ssd_mobilenet_v1_coco.pbtxt", processNet("dnn/ssd_mobilenet_v1_coco.pb", "dnn/ssd_mobilenet_v1_coco.pbtxt",
......
...@@ -279,7 +279,7 @@ TEST(Test_TensorFlow, Inception_v2_SSD) ...@@ -279,7 +279,7 @@ TEST(Test_TensorFlow, Inception_v2_SSD)
normAssert(detections, ref); normAssert(detections, ref);
} }
OCL_TEST(Test_TensorFlow, DISABLED_MobileNet_SSD) OCL_TEST(Test_TensorFlow, MobileNet_SSD)
{ {
std::string netPath = findDataFile("dnn/ssd_mobilenet_v1_coco.pb", false); std::string netPath = findDataFile("dnn/ssd_mobilenet_v1_coco.pb", false);
std::string netConfig = findDataFile("dnn/ssd_mobilenet_v1_coco.pbtxt", false); std::string netConfig = findDataFile("dnn/ssd_mobilenet_v1_coco.pbtxt", false);
...@@ -311,8 +311,8 @@ OCL_TEST(Test_TensorFlow, DISABLED_MobileNet_SSD) ...@@ -311,8 +311,8 @@ OCL_TEST(Test_TensorFlow, DISABLED_MobileNet_SSD)
std::vector<Mat> output; std::vector<Mat> output;
net.forward(output, outNames); net.forward(output, outNames);
normAssert(target[0].reshape(1, 1), output[0].reshape(1, 1)); normAssert(target[0].reshape(1, 1), output[0].reshape(1, 1), "", 1e-5, 1.5e-4);
normAssert(target[1].reshape(1, 1), output[1].reshape(1, 1), "", 1e-5, 2e-4); normAssert(target[1].reshape(1, 1), output[1].reshape(1, 1), "", 1e-5, 3e-4);
normAssert(target[2].reshape(1, 1), output[2].reshape(1, 1), "", 4e-5, 1e-2); normAssert(target[2].reshape(1, 1), output[2].reshape(1, 1), "", 4e-5, 1e-2);
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment