|
|
|
@ -98,6 +98,7 @@ public: |
|
|
|
|
|
|
|
|
|
TEST_P(DNNTestNetwork, AlexNet) |
|
|
|
|
{ |
|
|
|
|
applyTestTag(CV_TEST_TAG_MEMORY_1GB); |
|
|
|
|
processNet("dnn/bvlc_alexnet.caffemodel", "dnn/bvlc_alexnet.prototxt", |
|
|
|
|
Size(227, 227), "prob", |
|
|
|
|
target == DNN_TARGET_OPENCL ? "dnn/halide_scheduler_opencl_alexnet.yml" : |
|
|
|
@ -106,6 +107,7 @@ TEST_P(DNNTestNetwork, AlexNet) |
|
|
|
|
|
|
|
|
|
TEST_P(DNNTestNetwork, ResNet_50) |
|
|
|
|
{ |
|
|
|
|
applyTestTag(target == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_512MB : CV_TEST_TAG_MEMORY_1GB); |
|
|
|
|
processNet("dnn/ResNet-50-model.caffemodel", "dnn/ResNet-50-deploy.prototxt", |
|
|
|
|
Size(224, 224), "prob", |
|
|
|
|
target == DNN_TARGET_OPENCL ? "dnn/halide_scheduler_opencl_resnet_50.yml" : |
|
|
|
@ -122,12 +124,14 @@ TEST_P(DNNTestNetwork, SqueezeNet_v1_1) |
|
|
|
|
|
|
|
|
|
TEST_P(DNNTestNetwork, GoogLeNet) |
|
|
|
|
{ |
|
|
|
|
applyTestTag(target == DNN_TARGET_CPU ? "" : CV_TEST_TAG_MEMORY_512MB); |
|
|
|
|
processNet("dnn/bvlc_googlenet.caffemodel", "dnn/bvlc_googlenet.prototxt", |
|
|
|
|
Size(224, 224), "prob"); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
TEST_P(DNNTestNetwork, Inception_5h) |
|
|
|
|
{ |
|
|
|
|
applyTestTag(CV_TEST_TAG_MEMORY_512MB); |
|
|
|
|
double l1 = default_l1, lInf = default_lInf; |
|
|
|
|
if (backend == DNN_BACKEND_INFERENCE_ENGINE && (target == DNN_TARGET_CPU || target == DNN_TARGET_OPENCL)) |
|
|
|
|
{ |
|
|
|
@ -142,6 +146,7 @@ TEST_P(DNNTestNetwork, Inception_5h) |
|
|
|
|
|
|
|
|
|
TEST_P(DNNTestNetwork, ENet) |
|
|
|
|
{ |
|
|
|
|
applyTestTag(target == DNN_TARGET_CPU ? "" : CV_TEST_TAG_MEMORY_512MB); |
|
|
|
|
if ((backend == DNN_BACKEND_INFERENCE_ENGINE) || |
|
|
|
|
(backend == DNN_BACKEND_OPENCV && target == DNN_TARGET_OPENCL_FP16)) |
|
|
|
|
throw SkipTestException(""); |
|
|
|
@ -153,6 +158,7 @@ TEST_P(DNNTestNetwork, ENet) |
|
|
|
|
|
|
|
|
|
TEST_P(DNNTestNetwork, MobileNet_SSD_Caffe) |
|
|
|
|
{ |
|
|
|
|
applyTestTag(CV_TEST_TAG_MEMORY_512MB); |
|
|
|
|
if (backend == DNN_BACKEND_HALIDE) |
|
|
|
|
throw SkipTestException(""); |
|
|
|
|
Mat sample = imread(findDataFile("dnn/street.png", false)); |
|
|
|
@ -184,6 +190,7 @@ TEST_P(DNNTestNetwork, MobileNet_SSD_Caffe_Different_Width_Height) |
|
|
|
|
|
|
|
|
|
TEST_P(DNNTestNetwork, MobileNet_SSD_v1_TensorFlow) |
|
|
|
|
{ |
|
|
|
|
applyTestTag(target == DNN_TARGET_CPU ? "" : CV_TEST_TAG_MEMORY_512MB); |
|
|
|
|
if (backend == DNN_BACKEND_HALIDE) |
|
|
|
|
throw SkipTestException(""); |
|
|
|
|
Mat sample = imread(findDataFile("dnn/street.png", false)); |
|
|
|
@ -214,6 +221,7 @@ TEST_P(DNNTestNetwork, MobileNet_SSD_v1_TensorFlow_Different_Width_Height) |
|
|
|
|
|
|
|
|
|
TEST_P(DNNTestNetwork, MobileNet_SSD_v2_TensorFlow) |
|
|
|
|
{ |
|
|
|
|
applyTestTag(target == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_512MB : CV_TEST_TAG_MEMORY_1GB); |
|
|
|
|
if (backend == DNN_BACKEND_HALIDE) |
|
|
|
|
throw SkipTestException(""); |
|
|
|
|
Mat sample = imread(findDataFile("dnn/street.png", false)); |
|
|
|
@ -226,6 +234,8 @@ TEST_P(DNNTestNetwork, MobileNet_SSD_v2_TensorFlow) |
|
|
|
|
|
|
|
|
|
TEST_P(DNNTestNetwork, SSD_VGG16) |
|
|
|
|
{ |
|
|
|
|
applyTestTag(CV_TEST_TAG_LONG, (target == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_1GB : CV_TEST_TAG_MEMORY_2GB), |
|
|
|
|
CV_TEST_TAG_DEBUG_VERYLONG); |
|
|
|
|
if (backend == DNN_BACKEND_HALIDE && target == DNN_TARGET_CPU) |
|
|
|
|
throw SkipTestException(""); |
|
|
|
|
double scoreThreshold = (target == DNN_TARGET_OPENCL_FP16 || target == DNN_TARGET_MYRIAD) ? 0.0325 : 0.0; |
|
|
|
@ -238,6 +248,8 @@ TEST_P(DNNTestNetwork, SSD_VGG16) |
|
|
|
|
|
|
|
|
|
TEST_P(DNNTestNetwork, OpenPose_pose_coco) |
|
|
|
|
{ |
|
|
|
|
applyTestTag(CV_TEST_TAG_LONG, (target == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_1GB : CV_TEST_TAG_MEMORY_2GB), |
|
|
|
|
CV_TEST_TAG_DEBUG_VERYLONG); |
|
|
|
|
if (backend == DNN_BACKEND_HALIDE) |
|
|
|
|
throw SkipTestException(""); |
|
|
|
|
#if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_LE(2018050000) |
|
|
|
@ -254,6 +266,8 @@ TEST_P(DNNTestNetwork, OpenPose_pose_coco) |
|
|
|
|
|
|
|
|
|
TEST_P(DNNTestNetwork, OpenPose_pose_mpi) |
|
|
|
|
{ |
|
|
|
|
applyTestTag(CV_TEST_TAG_LONG, (target == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_1GB : CV_TEST_TAG_MEMORY_2GB), |
|
|
|
|
CV_TEST_TAG_DEBUG_VERYLONG); |
|
|
|
|
if (backend == DNN_BACKEND_HALIDE) |
|
|
|
|
throw SkipTestException(""); |
|
|
|
|
#if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_LE(2018050000) |
|
|
|
@ -270,6 +284,7 @@ TEST_P(DNNTestNetwork, OpenPose_pose_mpi) |
|
|
|
|
|
|
|
|
|
TEST_P(DNNTestNetwork, OpenPose_pose_mpi_faster_4_stages) |
|
|
|
|
{ |
|
|
|
|
applyTestTag(CV_TEST_TAG_LONG, CV_TEST_TAG_MEMORY_1GB); |
|
|
|
|
if (backend == DNN_BACKEND_HALIDE) |
|
|
|
|
throw SkipTestException(""); |
|
|
|
|
#if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_LE(2018050000) |
|
|
|
@ -318,6 +333,7 @@ TEST_P(DNNTestNetwork, opencv_face_detector) |
|
|
|
|
|
|
|
|
|
TEST_P(DNNTestNetwork, Inception_v2_SSD_TensorFlow) |
|
|
|
|
{ |
|
|
|
|
applyTestTag(target == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_512MB : CV_TEST_TAG_MEMORY_1GB); |
|
|
|
|
#if defined(INF_ENGINE_RELEASE) |
|
|
|
|
if (backend == DNN_BACKEND_INFERENCE_ENGINE && target == DNN_TARGET_MYRIAD |
|
|
|
|
&& getInferenceEngineVPUType() == CV_DNN_INFERENCE_ENGINE_VPU_TYPE_MYRIAD_X) |
|
|
|
@ -335,6 +351,7 @@ TEST_P(DNNTestNetwork, Inception_v2_SSD_TensorFlow) |
|
|
|
|
|
|
|
|
|
TEST_P(DNNTestNetwork, DenseNet_121) |
|
|
|
|
{ |
|
|
|
|
applyTestTag(CV_TEST_TAG_MEMORY_512MB); |
|
|
|
|
if (backend == DNN_BACKEND_HALIDE) |
|
|
|
|
throw SkipTestException(""); |
|
|
|
|
// Reference output values are in range [-3.807, 4.605]
|
|
|
|
|