|
|
@ -101,6 +101,9 @@ public: |
|
|
|
TEST_P(DNNTestNetwork, AlexNet) |
|
|
|
TEST_P(DNNTestNetwork, AlexNet) |
|
|
|
{ |
|
|
|
{ |
|
|
|
applyTestTag(CV_TEST_TAG_MEMORY_1GB); |
|
|
|
applyTestTag(CV_TEST_TAG_MEMORY_1GB); |
|
|
|
|
|
|
|
if (backend == DNN_BACKEND_HALIDE) // Realization contains wrong number of Images (1) for realizing pipeline with 2 outputs
|
|
|
|
|
|
|
|
applyTestTag(CV_TEST_TAG_DNN_SKIP_HALIDE); |
|
|
|
|
|
|
|
|
|
|
|
processNet("dnn/bvlc_alexnet.caffemodel", "dnn/bvlc_alexnet.prototxt", |
|
|
|
processNet("dnn/bvlc_alexnet.caffemodel", "dnn/bvlc_alexnet.prototxt", |
|
|
|
Size(227, 227), "prob", |
|
|
|
Size(227, 227), "prob", |
|
|
|
target == DNN_TARGET_OPENCL ? "dnn/halide_scheduler_opencl_alexnet.yml" : |
|
|
|
target == DNN_TARGET_OPENCL ? "dnn/halide_scheduler_opencl_alexnet.yml" : |
|
|
@ -114,6 +117,9 @@ TEST_P(DNNTestNetwork, ResNet_50) |
|
|
|
(target == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_512MB : CV_TEST_TAG_MEMORY_1GB), |
|
|
|
(target == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_512MB : CV_TEST_TAG_MEMORY_1GB), |
|
|
|
CV_TEST_TAG_DEBUG_LONG |
|
|
|
CV_TEST_TAG_DEBUG_LONG |
|
|
|
); |
|
|
|
); |
|
|
|
|
|
|
|
if (backend == DNN_BACKEND_HALIDE) // Realization contains wrong number of Images (1) for realizing pipeline with 2 outputs
|
|
|
|
|
|
|
|
applyTestTag(CV_TEST_TAG_DNN_SKIP_HALIDE); |
|
|
|
|
|
|
|
|
|
|
|
processNet("dnn/ResNet-50-model.caffemodel", "dnn/ResNet-50-deploy.prototxt", |
|
|
|
processNet("dnn/ResNet-50-model.caffemodel", "dnn/ResNet-50-deploy.prototxt", |
|
|
|
Size(224, 224), "prob", |
|
|
|
Size(224, 224), "prob", |
|
|
|
target == DNN_TARGET_OPENCL ? "dnn/halide_scheduler_opencl_resnet_50.yml" : |
|
|
|
target == DNN_TARGET_OPENCL ? "dnn/halide_scheduler_opencl_resnet_50.yml" : |
|
|
@ -123,6 +129,9 @@ TEST_P(DNNTestNetwork, ResNet_50) |
|
|
|
|
|
|
|
|
|
|
|
TEST_P(DNNTestNetwork, SqueezeNet_v1_1) |
|
|
|
TEST_P(DNNTestNetwork, SqueezeNet_v1_1) |
|
|
|
{ |
|
|
|
{ |
|
|
|
|
|
|
|
if (backend == DNN_BACKEND_HALIDE) // Realization contains wrong number of Images (1) for realizing pipeline with 2 outputs
|
|
|
|
|
|
|
|
applyTestTag(CV_TEST_TAG_DNN_SKIP_HALIDE); |
|
|
|
|
|
|
|
|
|
|
|
processNet("dnn/squeezenet_v1.1.caffemodel", "dnn/squeezenet_v1.1.prototxt", |
|
|
|
processNet("dnn/squeezenet_v1.1.caffemodel", "dnn/squeezenet_v1.1.prototxt", |
|
|
|
Size(227, 227), "prob", |
|
|
|
Size(227, 227), "prob", |
|
|
|
target == DNN_TARGET_OPENCL ? "dnn/halide_scheduler_opencl_squeezenet_v1_1.yml" : |
|
|
|
target == DNN_TARGET_OPENCL ? "dnn/halide_scheduler_opencl_squeezenet_v1_1.yml" : |
|
|
@ -133,6 +142,9 @@ TEST_P(DNNTestNetwork, SqueezeNet_v1_1) |
|
|
|
TEST_P(DNNTestNetwork, GoogLeNet) |
|
|
|
TEST_P(DNNTestNetwork, GoogLeNet) |
|
|
|
{ |
|
|
|
{ |
|
|
|
applyTestTag(target == DNN_TARGET_CPU ? "" : CV_TEST_TAG_MEMORY_512MB); |
|
|
|
applyTestTag(target == DNN_TARGET_CPU ? "" : CV_TEST_TAG_MEMORY_512MB); |
|
|
|
|
|
|
|
if (backend == DNN_BACKEND_HALIDE) // Realization contains wrong number of Images (1) for realizing pipeline with 2 outputs
|
|
|
|
|
|
|
|
applyTestTag(CV_TEST_TAG_DNN_SKIP_HALIDE); |
|
|
|
|
|
|
|
|
|
|
|
processNet("dnn/bvlc_googlenet.caffemodel", "dnn/bvlc_googlenet.prototxt", |
|
|
|
processNet("dnn/bvlc_googlenet.caffemodel", "dnn/bvlc_googlenet.prototxt", |
|
|
|
Size(224, 224), "prob"); |
|
|
|
Size(224, 224), "prob"); |
|
|
|
expectNoFallbacksFromIE(net); |
|
|
|
expectNoFallbacksFromIE(net); |
|
|
@ -141,6 +153,9 @@ TEST_P(DNNTestNetwork, GoogLeNet) |
|
|
|
TEST_P(DNNTestNetwork, Inception_5h) |
|
|
|
TEST_P(DNNTestNetwork, Inception_5h) |
|
|
|
{ |
|
|
|
{ |
|
|
|
applyTestTag(CV_TEST_TAG_MEMORY_512MB); |
|
|
|
applyTestTag(CV_TEST_TAG_MEMORY_512MB); |
|
|
|
|
|
|
|
if (backend == DNN_BACKEND_HALIDE) // Realization contains wrong number of Images (1) for realizing pipeline with 2 outputs
|
|
|
|
|
|
|
|
applyTestTag(CV_TEST_TAG_DNN_SKIP_HALIDE); |
|
|
|
|
|
|
|
|
|
|
|
double l1 = default_l1, lInf = default_lInf; |
|
|
|
double l1 = default_l1, lInf = default_lInf; |
|
|
|
if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 && (target == DNN_TARGET_CPU || target == DNN_TARGET_OPENCL)) |
|
|
|
if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 && (target == DNN_TARGET_CPU || target == DNN_TARGET_OPENCL)) |
|
|
|
{ |
|
|
|
{ |
|
|
@ -157,6 +172,9 @@ TEST_P(DNNTestNetwork, Inception_5h) |
|
|
|
TEST_P(DNNTestNetwork, ENet) |
|
|
|
TEST_P(DNNTestNetwork, ENet) |
|
|
|
{ |
|
|
|
{ |
|
|
|
applyTestTag(target == DNN_TARGET_CPU ? "" : CV_TEST_TAG_MEMORY_512MB); |
|
|
|
applyTestTag(target == DNN_TARGET_CPU ? "" : CV_TEST_TAG_MEMORY_512MB); |
|
|
|
|
|
|
|
if (backend == DNN_BACKEND_HALIDE) // Realization contains wrong number of Images (1) for realizing pipeline with 2 outputs
|
|
|
|
|
|
|
|
applyTestTag(CV_TEST_TAG_DNN_SKIP_HALIDE); |
|
|
|
|
|
|
|
|
|
|
|
if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019) |
|
|
|
if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019) |
|
|
|
applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER); |
|
|
|
applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER); |
|
|
|
if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH) |
|
|
|
if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH) |
|
|
|