2018-01-21 02:55:25 +08:00
// This file is part of OpenCV project.
// It is subject to the license terms in the LICENSE file found in the top-level directory
// of this distribution and at http://opencv.org/license.html.
//
2019-03-29 21:42:58 +08:00
// Copyright (C) 2018-2019, Intel Corporation, all rights reserved.
2018-01-21 02:55:25 +08:00
// Third party copyrights are property of their respective owners.
# include "test_precomp.hpp"
# include "opencv2/core/ocl.hpp"
2017-11-05 21:48:40 +08:00
namespace opencv_test { namespace {
2018-01-21 02:55:25 +08:00
2018-06-27 21:34:36 +08:00
class DNNTestNetwork : public DNNTestLayer
2018-01-21 02:55:25 +08:00
{
public :
void processNet ( const std : : string & weights , const std : : string & proto ,
2018-04-19 20:04:57 +08:00
Size inpSize , const std : : string & outputLayer = " " ,
2018-03-17 00:27:04 +08:00
const std : : string & halideScheduler = " " ,
2018-04-19 20:04:57 +08:00
double l1 = 0.0 , double lInf = 0.0 )
2018-01-21 02:55:25 +08:00
{
// Create a common input blob.
int blobSize [ ] = { 1 , 3 , inpSize . height , inpSize . width } ;
Mat inp ( 4 , blobSize , CV_32FC1 ) ;
randu ( inp , 0.0f , 1.0f ) ;
2018-03-17 00:27:04 +08:00
processNet ( weights , proto , inp , outputLayer , halideScheduler , l1 , lInf ) ;
2018-01-21 02:55:25 +08:00
}
void processNet ( std : : string weights , std : : string proto ,
2018-04-19 20:04:57 +08:00
Mat inp , const std : : string & outputLayer = " " ,
2018-03-17 00:27:04 +08:00
std : : string halideScheduler = " " ,
2018-06-08 21:55:21 +08:00
double l1 = 0.0 , double lInf = 0.0 , double detectionConfThresh = 0.2 )
2018-01-21 02:55:25 +08:00
{
2018-06-27 21:34:36 +08:00
checkBackend ( ) ;
l1 = l1 ? l1 : default_l1 ;
lInf = lInf ? lInf : default_lInf ;
2018-01-21 02:55:25 +08:00
weights = findDataFile ( weights , false ) ;
if ( ! proto . empty ( ) )
2019-06-20 21:43:28 +08:00
proto = findDataFile ( proto ) ;
2018-01-21 02:55:25 +08:00
// Create two networks - with default backend and target and a tested one.
2018-03-17 00:27:04 +08:00
Net netDefault = readNet ( weights , proto ) ;
2018-06-01 15:54:12 +08:00
netDefault . setPreferableBackend ( DNN_BACKEND_OPENCV ) ;
2018-01-21 02:55:25 +08:00
netDefault . setInput ( inp ) ;
Mat outDefault = netDefault . forward ( outputLayer ) . clone ( ) ;
2019-04-19 19:54:08 +08:00
net = readNet ( weights , proto ) ;
2018-01-21 02:55:25 +08:00
net . setInput ( inp ) ;
net . setPreferableBackend ( backend ) ;
net . setPreferableTarget ( target ) ;
if ( backend = = DNN_BACKEND_HALIDE & & ! halideScheduler . empty ( ) )
{
2019-06-20 21:43:28 +08:00
halideScheduler = findDataFile ( halideScheduler ) ;
2018-01-21 02:55:25 +08:00
net . setHalideScheduler ( halideScheduler ) ;
}
Mat out = net . forward ( outputLayer ) . clone ( ) ;
2018-06-08 21:55:21 +08:00
check ( outDefault , out , outputLayer , l1 , lInf , detectionConfThresh , " First run " ) ;
2018-01-21 02:55:25 +08:00
// Test 2: change input.
2018-04-19 20:04:57 +08:00
float * inpData = ( float * ) inp . data ;
for ( int i = 0 ; i < inp . size [ 0 ] * inp . size [ 1 ] ; + + i )
{
Mat slice ( inp . size [ 2 ] , inp . size [ 3 ] , CV_32F , inpData ) ;
cv : : flip ( slice , slice , 1 ) ;
inpData + = slice . total ( ) ;
}
2018-01-21 02:55:25 +08:00
netDefault . setInput ( inp ) ;
net . setInput ( inp ) ;
outDefault = netDefault . forward ( outputLayer ) . clone ( ) ;
out = net . forward ( outputLayer ) . clone ( ) ;
2018-06-08 21:55:21 +08:00
check ( outDefault , out , outputLayer , l1 , lInf , detectionConfThresh , " Second run " ) ;
2018-05-31 19:05:21 +08:00
}
2018-01-21 02:55:25 +08:00
2018-06-08 21:55:21 +08:00
void check ( Mat & ref , Mat & out , const std : : string & outputLayer , double l1 , double lInf ,
double detectionConfThresh , const char * msg )
2018-05-31 19:05:21 +08:00
{
2018-01-21 02:55:25 +08:00
if ( outputLayer = = " detection_out " )
2018-05-31 19:05:21 +08:00
{
2019-12-02 21:16:06 +08:00
if ( backend = = DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 )
2018-05-31 19:05:21 +08:00
{
// Inference Engine produces detections terminated by a row which starts from -1.
out = out . reshape ( 1 , out . total ( ) / 7 ) ;
int numDetections = 0 ;
while ( numDetections < out . rows & & out . at < float > ( numDetections , 0 ) ! = - 1 )
{
numDetections + = 1 ;
}
out = out . rowRange ( 0 , numDetections ) ;
}
2018-06-08 21:55:21 +08:00
normAssertDetections ( ref , out , msg , detectionConfThresh , l1 , lInf ) ;
2018-05-31 19:05:21 +08:00
}
2018-01-21 02:55:25 +08:00
else
2018-05-31 19:05:21 +08:00
normAssert ( ref , out , msg , l1 , lInf ) ;
2018-01-21 02:55:25 +08:00
}
2019-04-19 19:54:08 +08:00
Net net ;
2018-01-21 02:55:25 +08:00
} ;
TEST_P ( DNNTestNetwork , AlexNet )
{
2018-10-09 06:38:06 +08:00
applyTestTag ( CV_TEST_TAG_MEMORY_1GB ) ;
2020-12-11 02:23:24 +08:00
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 ) ;
2018-01-21 02:55:25 +08:00
processNet ( " dnn/bvlc_alexnet.caffemodel " , " dnn/bvlc_alexnet.prototxt " ,
2018-03-17 00:27:04 +08:00
Size ( 227 , 227 ) , " prob " ,
2018-01-21 02:55:25 +08:00
target = = DNN_TARGET_OPENCL ? " dnn/halide_scheduler_opencl_alexnet.yml " :
" dnn/halide_scheduler_alexnet.yml " ) ;
2019-04-19 19:54:08 +08:00
expectNoFallbacksFromIE ( net ) ;
2018-01-21 02:55:25 +08:00
}
TEST_P ( DNNTestNetwork , ResNet_50 )
{
2019-05-27 20:14:18 +08:00
applyTestTag (
( target = = DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_512MB : CV_TEST_TAG_MEMORY_1GB ) ,
CV_TEST_TAG_DEBUG_LONG
) ;
2020-12-11 02:23:24 +08:00
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 ) ;
2018-01-21 02:55:25 +08:00
processNet ( " dnn/ResNet-50-model.caffemodel " , " dnn/ResNet-50-deploy.prototxt " ,
2018-03-17 00:27:04 +08:00
Size ( 224 , 224 ) , " prob " ,
2018-01-21 02:55:25 +08:00
target = = DNN_TARGET_OPENCL ? " dnn/halide_scheduler_opencl_resnet_50.yml " :
" dnn/halide_scheduler_resnet_50.yml " ) ;
2019-04-19 19:54:08 +08:00
expectNoFallbacksFromIE ( net ) ;
2018-01-21 02:55:25 +08:00
}
TEST_P ( DNNTestNetwork , SqueezeNet_v1_1 )
{
2020-12-11 02:23:24 +08:00
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 ) ;
2018-01-21 02:55:25 +08:00
processNet ( " dnn/squeezenet_v1.1.caffemodel " , " dnn/squeezenet_v1.1.prototxt " ,
2018-03-17 00:27:04 +08:00
Size ( 227 , 227 ) , " prob " ,
2018-01-21 02:55:25 +08:00
target = = DNN_TARGET_OPENCL ? " dnn/halide_scheduler_opencl_squeezenet_v1_1.yml " :
" dnn/halide_scheduler_squeezenet_v1_1.yml " ) ;
2019-04-19 19:54:08 +08:00
expectNoFallbacksFromIE ( net ) ;
2018-01-21 02:55:25 +08:00
}
TEST_P ( DNNTestNetwork , GoogLeNet )
{
2018-10-09 06:38:06 +08:00
applyTestTag ( target = = DNN_TARGET_CPU ? " " : CV_TEST_TAG_MEMORY_512MB ) ;
2020-12-11 02:23:24 +08:00
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 ) ;
2018-01-21 02:55:25 +08:00
processNet ( " dnn/bvlc_googlenet.caffemodel " , " dnn/bvlc_googlenet.prototxt " ,
2018-03-17 00:27:04 +08:00
Size ( 224 , 224 ) , " prob " ) ;
2019-04-19 19:54:08 +08:00
expectNoFallbacksFromIE ( net ) ;
2018-01-21 02:55:25 +08:00
}
TEST_P ( DNNTestNetwork , Inception_5h )
{
2018-10-09 06:38:06 +08:00
applyTestTag ( CV_TEST_TAG_MEMORY_512MB ) ;
2020-12-11 02:23:24 +08:00
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 ) ;
2018-11-26 17:09:50 +08:00
double l1 = default_l1 , lInf = default_lInf ;
2019-12-02 21:16:06 +08:00
if ( backend = = DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 & & ( target = = DNN_TARGET_CPU | | target = = DNN_TARGET_OPENCL ) )
2018-11-26 17:09:50 +08:00
{
l1 = 1.72e-5 ;
lInf = 8e-4 ;
}
2018-03-17 00:27:04 +08:00
processNet ( " dnn/tensorflow_inception_graph.pb " , " " , Size ( 224 , 224 ) , " softmax2 " ,
2018-01-21 02:55:25 +08:00
target = = DNN_TARGET_OPENCL ? " dnn/halide_scheduler_opencl_inception_5h.yml " :
2018-11-26 17:09:50 +08:00
" dnn/halide_scheduler_inception_5h.yml " ,
l1 , lInf ) ;
2019-04-19 19:54:08 +08:00
expectNoFallbacksFromIE ( net ) ;
2018-01-21 02:55:25 +08:00
}
TEST_P ( DNNTestNetwork , ENet )
{
2018-10-09 06:38:06 +08:00
applyTestTag ( target = = DNN_TARGET_CPU ? " " : CV_TEST_TAG_MEMORY_512MB ) ;
2020-12-11 02:23:24 +08:00
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 ) ;
2019-12-02 21:16:06 +08:00
if ( backend = = DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 )
applyTestTag ( CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER ) ;
if ( backend = = DNN_BACKEND_INFERENCE_ENGINE_NGRAPH )
applyTestTag ( CV_TEST_TAG_DNN_SKIP_IE_NGRAPH ) ;
2019-06-15 20:17:25 +08:00
if ( backend = = DNN_BACKEND_OPENCV & & target = = DNN_TARGET_OPENCL_FP16 )
applyTestTag ( CV_TEST_TAG_DNN_SKIP_OPENCL_FP16 ) ;
2018-03-17 00:27:04 +08:00
processNet ( " dnn/Enet-model-best.net " , " " , Size ( 512 , 512 ) , " l367_Deconvolution " ,
2018-01-21 02:55:25 +08:00
target = = DNN_TARGET_OPENCL ? " dnn/halide_scheduler_opencl_enet.yml " :
" dnn/halide_scheduler_enet.yml " ,
2e-5 , 0.15 ) ;
}
2018-02-07 16:28:45 +08:00
TEST_P ( DNNTestNetwork , MobileNet_SSD_Caffe )
2018-01-21 02:55:25 +08:00
{
2018-10-09 06:38:06 +08:00
applyTestTag ( CV_TEST_TAG_MEMORY_512MB ) ;
2018-05-31 19:05:21 +08:00
if ( backend = = DNN_BACKEND_HALIDE )
2019-06-15 20:17:25 +08:00
applyTestTag ( CV_TEST_TAG_DNN_SKIP_HALIDE ) ;
2019-06-20 21:43:28 +08:00
Mat sample = imread ( findDataFile ( " dnn/street.png " ) ) ;
2018-01-21 02:55:25 +08:00
Mat inp = blobFromImage ( sample , 1.0f / 127.5 , Size ( 300 , 300 ) , Scalar ( 127.5 , 127.5 , 127.5 ) , false ) ;
2019-03-29 21:42:58 +08:00
float diffScores = ( target = = DNN_TARGET_OPENCL_FP16 | | target = = DNN_TARGET_MYRIAD ) ? 1.5e-2 : 0.0 ;
float diffSquares = ( target = = DNN_TARGET_MYRIAD ) ? 0.063 : 0.0 ;
2021-06-30 04:25:22 +08:00
float detectionConfThresh = ( target = = DNN_TARGET_MYRIAD ) ? 0.262 : FLT_MIN ;
2019-03-29 21:42:58 +08:00
processNet ( " dnn/MobileNetSSD_deploy.caffemodel " , " dnn/MobileNetSSD_deploy.prototxt " ,
inp , " detection_out " , " " , diffScores , diffSquares , detectionConfThresh ) ;
2019-04-19 19:54:08 +08:00
expectNoFallbacksFromIE ( net ) ;
2018-01-21 02:55:25 +08:00
}
2019-02-02 01:23:51 +08:00
TEST_P ( DNNTestNetwork , MobileNet_SSD_Caffe_Different_Width_Height )
{
if ( backend = = DNN_BACKEND_HALIDE )
2019-06-15 20:17:25 +08:00
applyTestTag ( CV_TEST_TAG_DNN_SKIP_HALIDE ) ;
2019-03-29 21:42:58 +08:00
# if defined(INF_ENGINE_RELEASE)
2019-12-24 18:34:33 +08:00
if ( ( backend = = DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 | | backend = = DNN_BACKEND_INFERENCE_ENGINE_NGRAPH ) & &
target = = DNN_TARGET_MYRIAD & & getInferenceEngineVPUType ( ) = = CV_DNN_INFERENCE_ENGINE_VPU_TYPE_MYRIAD_X )
2021-11-26 03:56:27 +08:00
applyTestTag ( CV_TEST_TAG_DNN_SKIP_IE_MYRIAD_X , CV_TEST_TAG_DNN_SKIP_IE_NGRAPH , CV_TEST_TAG_DNN_SKIP_IE_VERSION ) ;
# endif
# if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_EQ(2021040000)
// IE exception: Ngraph operation Transpose with name conv15_2_mbox_conf_perm has dynamic output shape on 0 port, but CPU plug-in supports only static shape
if ( backend = = DNN_BACKEND_INFERENCE_ENGINE_NGRAPH & & ( target = = DNN_TARGET_OPENCL | | target = = DNN_TARGET_OPENCL_FP16 ) )
applyTestTag ( target = = DNN_TARGET_OPENCL ? CV_TEST_TAG_DNN_SKIP_IE_OPENCL : CV_TEST_TAG_DNN_SKIP_IE_OPENCL_FP16 ,
CV_TEST_TAG_DNN_SKIP_IE_NGRAPH , CV_TEST_TAG_DNN_SKIP_IE_VERSION
) ;
2019-03-29 21:42:58 +08:00
# endif
2021-11-26 03:56:27 +08:00
2019-06-20 21:43:28 +08:00
Mat sample = imread ( findDataFile ( " dnn/street.png " ) ) ;
2019-02-02 01:23:51 +08:00
Mat inp = blobFromImage ( sample , 1.0f / 127.5 , Size ( 300 , 560 ) , Scalar ( 127.5 , 127.5 , 127.5 ) , false ) ;
float diffScores = ( target = = DNN_TARGET_OPENCL_FP16 | | target = = DNN_TARGET_MYRIAD ) ? 0.029 : 0.0 ;
float diffSquares = ( target = = DNN_TARGET_OPENCL_FP16 | | target = = DNN_TARGET_MYRIAD ) ? 0.09 : 0.0 ;
processNet ( " dnn/MobileNetSSD_deploy.caffemodel " , " dnn/MobileNetSSD_deploy.prototxt " ,
2019-03-29 21:42:58 +08:00
inp , " detection_out " , " " , diffScores , diffSquares ) ;
2019-04-19 19:54:08 +08:00
expectNoFallbacksFromIE ( net ) ;
2019-02-02 01:23:51 +08:00
}
2018-06-08 21:55:21 +08:00
TEST_P ( DNNTestNetwork , MobileNet_SSD_v1_TensorFlow )
2018-02-07 16:28:45 +08:00
{
2018-10-09 06:38:06 +08:00
applyTestTag ( target = = DNN_TARGET_CPU ? " " : CV_TEST_TAG_MEMORY_512MB ) ;
2018-06-08 21:55:21 +08:00
if ( backend = = DNN_BACKEND_HALIDE )
2019-06-15 20:17:25 +08:00
applyTestTag ( CV_TEST_TAG_DNN_SKIP_HALIDE ) ;
2019-07-25 14:57:49 +08:00
2019-06-20 21:43:28 +08:00
Mat sample = imread ( findDataFile ( " dnn/street.png " ) ) ;
2018-08-31 20:41:56 +08:00
Mat inp = blobFromImage ( sample , 1.0f , Size ( 300 , 300 ) , Scalar ( ) , false ) ;
2019-03-29 21:42:58 +08:00
float l1 = ( target = = DNN_TARGET_OPENCL_FP16 | | target = = DNN_TARGET_MYRIAD ) ? 0.095 : 0.0 ;
float lInf = ( target = = DNN_TARGET_OPENCL_FP16 | | target = = DNN_TARGET_MYRIAD ) ? 0.09 : 0.0 ;
float detectionConfThresh = ( target = = DNN_TARGET_MYRIAD ) ? 0.216 : 0.2 ;
2018-06-08 21:55:21 +08:00
processNet ( " dnn/ssd_mobilenet_v1_coco_2017_11_17.pb " , " dnn/ssd_mobilenet_v1_coco_2017_11_17.pbtxt " ,
2019-03-29 21:42:58 +08:00
inp , " detection_out " , " " , l1 , lInf , detectionConfThresh ) ;
2019-04-19 19:54:08 +08:00
expectNoFallbacksFromIE ( net ) ;
2019-02-02 01:23:51 +08:00
}
TEST_P ( DNNTestNetwork , MobileNet_SSD_v1_TensorFlow_Different_Width_Height )
{
if ( backend = = DNN_BACKEND_HALIDE )
2019-06-15 20:17:25 +08:00
applyTestTag ( CV_TEST_TAG_DNN_SKIP_HALIDE ) ;
2019-03-29 21:42:58 +08:00
# if defined(INF_ENGINE_RELEASE)
2019-12-24 18:34:33 +08:00
if ( ( backend = = DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 | | backend = = DNN_BACKEND_INFERENCE_ENGINE_NGRAPH ) & &
target = = DNN_TARGET_MYRIAD & & getInferenceEngineVPUType ( ) = = CV_DNN_INFERENCE_ENGINE_VPU_TYPE_MYRIAD_X )
2019-06-15 20:17:25 +08:00
applyTestTag ( CV_TEST_TAG_DNN_SKIP_IE_MYRIAD_X ) ;
2019-03-29 21:42:58 +08:00
# endif
2019-07-25 14:57:49 +08:00
# if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_EQ(2019020000)
2019-12-02 21:16:06 +08:00
if ( backend = = DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 & & target = = DNN_TARGET_MYRIAD )
applyTestTag ( CV_TEST_TAG_DNN_SKIP_IE_MYRIAD , CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER , CV_TEST_TAG_DNN_SKIP_IE_VERSION ) ;
2019-07-25 14:57:49 +08:00
# endif
2019-06-20 21:43:28 +08:00
Mat sample = imread ( findDataFile ( " dnn/street.png " ) ) ;
2019-02-02 01:23:51 +08:00
Mat inp = blobFromImage ( sample , 1.0f , Size ( 300 , 560 ) , Scalar ( ) , false ) ;
2020-02-26 22:51:18 +08:00
float l1 = ( target = = DNN_TARGET_OPENCL_FP16 | | target = = DNN_TARGET_MYRIAD ) ? 0.013 : 0.0 ;
2019-02-02 01:23:51 +08:00
float lInf = ( target = = DNN_TARGET_OPENCL_FP16 | | target = = DNN_TARGET_MYRIAD ) ? 0.06 : 0.0 ;
processNet ( " dnn/ssd_mobilenet_v1_coco_2017_11_17.pb " , " dnn/ssd_mobilenet_v1_coco_2017_11_17.pbtxt " ,
inp , " detection_out " , " " , l1 , lInf ) ;
2019-04-19 19:54:08 +08:00
expectNoFallbacksFromIE ( net ) ;
2018-02-07 16:28:45 +08:00
}
2018-06-08 21:55:21 +08:00
TEST_P ( DNNTestNetwork , MobileNet_SSD_v2_TensorFlow )
{
2018-10-09 06:38:06 +08:00
applyTestTag ( target = = DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_512MB : CV_TEST_TAG_MEMORY_1GB ) ;
2018-06-08 21:55:21 +08:00
if ( backend = = DNN_BACKEND_HALIDE )
2019-06-15 20:17:25 +08:00
applyTestTag ( CV_TEST_TAG_DNN_SKIP_HALIDE ) ;
2019-07-25 14:57:49 +08:00
2019-06-20 21:43:28 +08:00
Mat sample = imread ( findDataFile ( " dnn/street.png " ) ) ;
2018-08-31 20:41:56 +08:00
Mat inp = blobFromImage ( sample , 1.0f , Size ( 300 , 300 ) , Scalar ( ) , false ) ;
2019-01-14 14:55:44 +08:00
float l1 = ( target = = DNN_TARGET_OPENCL_FP16 | | target = = DNN_TARGET_MYRIAD ) ? 0.013 : 2e-5 ;
2018-07-31 16:37:45 +08:00
float lInf = ( target = = DNN_TARGET_OPENCL_FP16 | | target = = DNN_TARGET_MYRIAD ) ? 0.062 : 0.0 ;
2018-06-08 21:55:21 +08:00
processNet ( " dnn/ssd_mobilenet_v2_coco_2018_03_29.pb " , " dnn/ssd_mobilenet_v2_coco_2018_03_29.pbtxt " ,
inp , " detection_out " , " " , l1 , lInf , 0.25 ) ;
2019-04-19 19:54:08 +08:00
expectNoFallbacksFromIE ( net ) ;
2018-06-08 21:55:21 +08:00
}
2018-01-21 02:55:25 +08:00
TEST_P ( DNNTestNetwork , SSD_VGG16 )
{
2018-10-09 06:38:06 +08:00
applyTestTag ( CV_TEST_TAG_LONG , ( target = = DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_1GB : CV_TEST_TAG_MEMORY_2GB ) ,
CV_TEST_TAG_DEBUG_VERYLONG ) ;
2018-05-31 19:05:21 +08:00
if ( backend = = DNN_BACKEND_HALIDE & & target = = DNN_TARGET_CPU )
2019-06-15 20:17:25 +08:00
applyTestTag ( CV_TEST_TAG_DNN_SKIP_HALIDE ) ; // TODO HALIDE_CPU
2021-11-26 03:56:27 +08:00
2019-06-20 21:43:28 +08:00
Mat sample = imread ( findDataFile ( " dnn/street.png " ) ) ;
2018-05-31 19:05:21 +08:00
Mat inp = blobFromImage ( sample , 1.0f , Size ( 300 , 300 ) , Scalar ( ) , false ) ;
2021-11-26 03:56:27 +08:00
float scoreDiff = 0.0 , iouDiff = 0.0 ;
if ( target = = DNN_TARGET_OPENCL_FP16 )
{
scoreDiff = 0.04 ;
}
else if ( target = = DNN_TARGET_MYRIAD )
{
scoreDiff = 0.0325 ;
iouDiff = 0.032 ;
}
2018-01-21 02:55:25 +08:00
processNet ( " dnn/VGG_ILSVRC2016_SSD_300x300_iter_440000.caffemodel " ,
2021-11-26 03:56:27 +08:00
" dnn/ssd_vgg16.prototxt " , inp , " detection_out " , " " , scoreDiff , iouDiff ) ;
2019-04-19 19:54:08 +08:00
expectNoFallbacksFromIE ( net ) ;
2018-01-21 02:55:25 +08:00
}
2018-02-06 16:57:35 +08:00
TEST_P ( DNNTestNetwork , OpenPose_pose_coco )
{
2018-10-09 06:38:06 +08:00
applyTestTag ( CV_TEST_TAG_LONG , ( target = = DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_1GB : CV_TEST_TAG_MEMORY_2GB ) ,
2019-06-15 20:17:25 +08:00
CV_TEST_TAG_DEBUG_LONG ) ;
2018-11-26 17:09:50 +08:00
if ( backend = = DNN_BACKEND_HALIDE )
2019-06-15 20:17:25 +08:00
applyTestTag ( CV_TEST_TAG_DNN_SKIP_HALIDE ) ;
2019-03-29 21:42:58 +08:00
# if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_LE(2018050000)
2019-12-02 21:16:06 +08:00
if ( backend = = DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 & & target = = DNN_TARGET_MYRIAD
2019-03-29 21:42:58 +08:00
& & getInferenceEngineVPUType ( ) = = CV_DNN_INFERENCE_ENGINE_VPU_TYPE_MYRIAD_X )
2019-12-02 21:16:06 +08:00
applyTestTag ( CV_TEST_TAG_DNN_SKIP_IE_MYRIAD_X , CV_TEST_TAG_DNN_SKIP_IE_VERSION ) ;
2019-03-29 21:42:58 +08:00
# endif
2021-06-30 04:25:22 +08:00
const float l1 = ( target = = DNN_TARGET_MYRIAD ) ? 0.009 : 0.0 ;
const float lInf = ( target = = DNN_TARGET_MYRIAD ) ? 0.09 : 0.0 ;
2018-02-06 16:57:35 +08:00
processNet ( " dnn/openpose_pose_coco.caffemodel " , " dnn/openpose_pose_coco.prototxt " ,
2019-03-29 21:42:58 +08:00
Size ( 46 , 46 ) , " " , " " , l1 , lInf ) ;
2019-04-19 19:54:08 +08:00
expectNoFallbacksFromIE ( net ) ;
2018-02-06 16:57:35 +08:00
}
TEST_P ( DNNTestNetwork , OpenPose_pose_mpi )
{
2018-10-09 06:38:06 +08:00
applyTestTag ( CV_TEST_TAG_LONG , ( target = = DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_1GB : CV_TEST_TAG_MEMORY_2GB ) ,
CV_TEST_TAG_DEBUG_VERYLONG ) ;
2018-11-26 17:09:50 +08:00
if ( backend = = DNN_BACKEND_HALIDE )
2019-06-15 20:17:25 +08:00
applyTestTag ( CV_TEST_TAG_DNN_SKIP_HALIDE ) ;
2019-03-29 21:42:58 +08:00
# if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_LE(2018050000)
2019-12-02 21:16:06 +08:00
if ( backend = = DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 & & target = = DNN_TARGET_MYRIAD
2019-03-29 21:42:58 +08:00
& & getInferenceEngineVPUType ( ) = = CV_DNN_INFERENCE_ENGINE_VPU_TYPE_MYRIAD_X )
2019-12-02 21:16:06 +08:00
applyTestTag ( CV_TEST_TAG_DNN_SKIP_IE_MYRIAD_X , CV_TEST_TAG_DNN_SKIP_IE_VERSION ) ;
2019-03-29 21:42:58 +08:00
# endif
2019-06-14 23:17:02 +08:00
2019-03-29 21:42:58 +08:00
// output range: [-0.001, 0.97]
2021-06-30 04:25:22 +08:00
const float l1 = ( target = = DNN_TARGET_MYRIAD ) ? 0.02 : 0.0 ;
const float lInf = ( target = = DNN_TARGET_MYRIAD | | target = = DNN_TARGET_OPENCL_FP16 ) ? 0.2 : 0.0 ;
2018-02-06 16:57:35 +08:00
processNet ( " dnn/openpose_pose_mpi.caffemodel " , " dnn/openpose_pose_mpi.prototxt " ,
2019-03-29 21:42:58 +08:00
Size ( 46 , 46 ) , " " , " " , l1 , lInf ) ;
2019-04-19 19:54:08 +08:00
expectNoFallbacksFromIE ( net ) ;
2018-02-06 16:57:35 +08:00
}
TEST_P ( DNNTestNetwork , OpenPose_pose_mpi_faster_4_stages )
{
2018-10-09 06:38:06 +08:00
applyTestTag ( CV_TEST_TAG_LONG , CV_TEST_TAG_MEMORY_1GB ) ;
2018-11-26 17:09:50 +08:00
if ( backend = = DNN_BACKEND_HALIDE )
2019-06-15 20:17:25 +08:00
applyTestTag ( CV_TEST_TAG_DNN_SKIP_HALIDE ) ;
2019-03-29 21:42:58 +08:00
# if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_LE(2018050000)
2019-12-02 21:16:06 +08:00
if ( backend = = DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 & & target = = DNN_TARGET_MYRIAD
2019-03-29 21:42:58 +08:00
& & getInferenceEngineVPUType ( ) = = CV_DNN_INFERENCE_ENGINE_VPU_TYPE_MYRIAD_X )
2019-12-02 21:16:06 +08:00
applyTestTag ( CV_TEST_TAG_DNN_SKIP_IE_MYRIAD_X , CV_TEST_TAG_DNN_SKIP_IE_VERSION ) ;
2019-03-29 21:42:58 +08:00
# endif
2019-06-14 23:17:02 +08:00
2018-02-06 16:57:35 +08:00
// The same .caffemodel but modified .prototxt
// See https://github.com/CMU-Perceptual-Computing-Lab/openpose/blob/master/src/openpose/pose/poseParameters.cpp
processNet ( " dnn/openpose_pose_mpi.caffemodel " , " dnn/openpose_pose_mpi_faster_4_stages.prototxt " ,
2018-11-23 23:23:27 +08:00
Size ( 46 , 46 ) ) ;
2019-04-19 19:54:08 +08:00
expectNoFallbacksFromIE ( net ) ;
2018-02-06 16:57:35 +08:00
}
TEST_P ( DNNTestNetwork , OpenFace )
{
2018-11-26 17:09:50 +08:00
# if defined(INF_ENGINE_RELEASE)
2019-03-29 21:42:58 +08:00
# if INF_ENGINE_VER_MAJOR_EQ(2018050000)
2019-12-02 21:16:06 +08:00
if ( backend = = DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 & & target = = DNN_TARGET_MYRIAD )
applyTestTag ( CV_TEST_TAG_DNN_SKIP_IE_MYRIAD , CV_TEST_TAG_DNN_SKIP_IE_VERSION ) ;
2018-11-26 17:09:50 +08:00
# endif
2018-08-27 20:45:44 +08:00
# endif
2018-11-26 17:09:50 +08:00
if ( backend = = DNN_BACKEND_HALIDE )
2019-06-15 20:17:25 +08:00
applyTestTag ( CV_TEST_TAG_DNN_SKIP_HALIDE ) ;
2019-03-29 21:42:58 +08:00
const float l1 = ( target = = DNN_TARGET_MYRIAD ) ? 0.0024 : 0.0 ;
const float lInf = ( target = = DNN_TARGET_MYRIAD ) ? 0.0071 : 0.0 ;
processNet ( " dnn/openface_nn4.small2.v1.t7 " , " " , Size ( 96 , 96 ) , " " , " " , l1 , lInf ) ;
2018-02-06 16:57:35 +08:00
}
2018-02-06 21:23:18 +08:00
TEST_P ( DNNTestNetwork , opencv_face_detector )
{
2018-05-31 19:05:21 +08:00
if ( backend = = DNN_BACKEND_HALIDE )
2019-06-15 20:17:25 +08:00
applyTestTag ( CV_TEST_TAG_DNN_SKIP_HALIDE ) ;
2019-06-20 21:43:28 +08:00
Mat img = imread ( findDataFile ( " gpu/lbpcascade/er.png " ) ) ;
2018-07-31 16:37:45 +08:00
Mat inp = blobFromImage ( img , 1.0 , Size ( ) , Scalar ( 104.0 , 177.0 , 123.0 ) , false , false ) ;
2018-02-06 21:23:18 +08:00
processNet ( " dnn/opencv_face_detector.caffemodel " , " dnn/opencv_face_detector.prototxt " ,
2018-03-17 00:27:04 +08:00
inp , " detection_out " ) ;
2019-04-19 19:54:08 +08:00
expectNoFallbacksFromIE ( net ) ;
2018-02-06 21:23:18 +08:00
}
2018-02-07 16:28:45 +08:00
TEST_P ( DNNTestNetwork , Inception_v2_SSD_TensorFlow )
{
2019-05-27 20:14:18 +08:00
applyTestTag (
( target = = DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_512MB : CV_TEST_TAG_MEMORY_1GB ) ,
CV_TEST_TAG_DEBUG_LONG
) ;
2019-03-29 21:42:58 +08:00
# if defined(INF_ENGINE_RELEASE)
2019-12-02 21:16:06 +08:00
if ( backend = = DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 & & target = = DNN_TARGET_MYRIAD
2019-03-29 21:42:58 +08:00
& & getInferenceEngineVPUType ( ) = = CV_DNN_INFERENCE_ENGINE_VPU_TYPE_MYRIAD_X )
2019-06-15 20:17:25 +08:00
applyTestTag ( CV_TEST_TAG_DNN_SKIP_IE_MYRIAD_X ) ;
2019-07-25 14:57:49 +08:00
# endif
# if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_EQ(2019020000)
2019-12-02 21:16:06 +08:00
if ( backend = = DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 & & target = = DNN_TARGET_MYRIAD )
applyTestTag ( CV_TEST_TAG_DNN_SKIP_IE_MYRIAD , CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER , CV_TEST_TAG_DNN_SKIP_IE_VERSION ) ;
2019-03-29 21:42:58 +08:00
# endif
2018-06-08 21:55:21 +08:00
if ( backend = = DNN_BACKEND_HALIDE )
2019-06-15 20:17:25 +08:00
applyTestTag ( CV_TEST_TAG_DNN_SKIP_HALIDE ) ;
2019-06-20 21:43:28 +08:00
Mat sample = imread ( findDataFile ( " dnn/street.png " ) ) ;
2018-08-31 20:41:56 +08:00
Mat inp = blobFromImage ( sample , 1.0f , Size ( 300 , 300 ) , Scalar ( ) , false ) ;
float l1 = ( target = = DNN_TARGET_OPENCL_FP16 | | target = = DNN_TARGET_MYRIAD ) ? 0.015 : 0.0 ;
2018-07-31 16:37:45 +08:00
float lInf = ( target = = DNN_TARGET_OPENCL_FP16 | | target = = DNN_TARGET_MYRIAD ) ? 0.0731 : 0.0 ;
2018-02-07 16:28:45 +08:00
processNet ( " dnn/ssd_inception_v2_coco_2017_11_17.pb " , " dnn/ssd_inception_v2_coco_2017_11_17.pbtxt " ,
2018-04-28 23:50:37 +08:00
inp , " detection_out " , " " , l1 , lInf ) ;
2019-04-19 19:54:08 +08:00
expectNoFallbacksFromIE ( net ) ;
2018-02-07 16:28:45 +08:00
}
2018-03-12 22:35:28 +08:00
TEST_P ( DNNTestNetwork , DenseNet_121 )
{
2018-10-09 06:38:06 +08:00
applyTestTag ( CV_TEST_TAG_MEMORY_512MB ) ;
2018-08-27 20:45:44 +08:00
if ( backend = = DNN_BACKEND_HALIDE )
2019-06-15 20:17:25 +08:00
applyTestTag ( CV_TEST_TAG_DNN_SKIP_HALIDE ) ;
2019-03-29 21:42:58 +08:00
// Reference output values are in range [-3.807, 4.605]
2018-08-27 20:45:44 +08:00
float l1 = 0.0 , lInf = 0.0 ;
if ( target = = DNN_TARGET_OPENCL_FP16 )
{
2019-07-27 18:30:15 +08:00
l1 = 2e-2 ; lInf = 9e-2 ;
2020-07-16 06:52:08 +08:00
if ( backend = = DNN_BACKEND_INFERENCE_ENGINE_NGRAPH )
lInf = 0.1f ;
2018-08-27 20:45:44 +08:00
}
else if ( target = = DNN_TARGET_MYRIAD )
{
2019-03-29 21:42:58 +08:00
l1 = 0.1 ; lInf = 0.6 ;
2018-08-27 20:45:44 +08:00
}
processNet ( " dnn/DenseNet_121.caffemodel " , " dnn/DenseNet_121.prototxt " , Size ( 224 , 224 ) , " " , " " , l1 , lInf ) ;
2019-06-25 02:55:32 +08:00
if ( target ! = DNN_TARGET_MYRIAD | | getInferenceEngineVPUType ( ) ! = CV_DNN_INFERENCE_ENGINE_VPU_TYPE_MYRIAD_X )
expectNoFallbacksFromIE ( net ) ;
2018-03-12 22:35:28 +08:00
}
2018-06-09 20:37:04 +08:00
TEST_P ( DNNTestNetwork , FastNeuralStyle_eccv16 )
{
2019-05-27 20:14:18 +08:00
applyTestTag ( CV_TEST_TAG_MEMORY_512MB , CV_TEST_TAG_DEBUG_VERYLONG ) ;
2019-06-15 20:17:25 +08:00
if ( backend = = DNN_BACKEND_HALIDE )
applyTestTag ( CV_TEST_TAG_DNN_SKIP_HALIDE ) ;
2019-12-02 21:16:06 +08:00
if ( backend = = DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 & & target = = DNN_TARGET_MYRIAD )
2020-02-07 21:40:50 +08:00
applyTestTag ( CV_TEST_TAG_DNN_SKIP_IE_MYRIAD , CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER ) ;
if ( backend = = DNN_BACKEND_INFERENCE_ENGINE_NGRAPH & & target = = DNN_TARGET_MYRIAD )
applyTestTag ( CV_TEST_TAG_DNN_SKIP_IE_MYRIAD , CV_TEST_TAG_DNN_SKIP_IE_NGRAPH ) ;
2019-02-08 22:12:33 +08:00
# if defined(INF_ENGINE_RELEASE)
2019-06-15 20:17:25 +08:00
# if INF_ENGINE_VER_MAJOR_LE(2018050000)
2019-12-02 21:16:06 +08:00
if ( backend = = DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 & & target = = DNN_TARGET_OPENCL )
applyTestTag ( CV_TEST_TAG_DNN_SKIP_IE_OPENCL , CV_TEST_TAG_DNN_SKIP_IE_VERSION ) ;
2019-02-08 22:12:33 +08:00
# endif
# endif
2019-06-20 21:43:28 +08:00
Mat img = imread ( findDataFile ( " dnn/googlenet_1.png " ) ) ;
2018-06-09 20:37:04 +08:00
Mat inp = blobFromImage ( img , 1.0 , Size ( 320 , 240 ) , Scalar ( 103.939 , 116.779 , 123.68 ) , false , false ) ;
// Output image has values in range [-143.526, 148.539].
2019-05-25 00:28:37 +08:00
float l1 = ( target = = DNN_TARGET_OPENCL_FP16 | | target = = DNN_TARGET_MYRIAD ) ? 0.4 : 4e-5 ;
2019-05-28 23:24:25 +08:00
float lInf = ( target = = DNN_TARGET_OPENCL_FP16 | | target = = DNN_TARGET_MYRIAD ) ? 7.45 : 2e-3 ;
2018-06-09 20:37:04 +08:00
processNet ( " dnn/fast_neural_style_eccv16_starry_night.t7 " , " " , inp , " " , " " , l1 , lInf ) ;
2019-06-15 20:17:25 +08:00
# if defined(HAVE_INF_ENGINE) && INF_ENGINE_VER_MAJOR_GE(2019010000)
2019-05-28 23:24:25 +08:00
expectNoFallbacksFromIE ( net ) ;
2019-06-14 23:17:02 +08:00
# endif
2018-06-09 20:37:04 +08:00
}
2018-12-05 23:31:14 +08:00
INSTANTIATE_TEST_CASE_P ( /*nothing*/ , DNNTestNetwork , dnnBackendsAndTargets ( true , true , false ) ) ;
2018-01-21 02:55:25 +08:00
2017-11-05 21:48:40 +08:00
} } // namespace