Skip to content

Commit e0841f3

Browse files
alalekalalek
committed
dnn(test-tags): add time / memory tags
1 parent b38de57 commit e0841f3

File tree

6 files changed

+65
-12
lines changed

6 files changed

+65
-12
lines changed

‎modules/dnn/test/test_backends.cpp

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -98,6 +98,7 @@ class DNNTestNetwork : public DNNTestLayer
9898

9999
TEST_P(DNNTestNetwork, AlexNet)
100100
{
101+
applyTestTag(CV_TEST_TAG_MEMORY_1GB);
101102
processNet("dnn/bvlc_alexnet.caffemodel", "dnn/bvlc_alexnet.prototxt",
102103
Size(227, 227), "prob",
103104
target == DNN_TARGET_OPENCL ? "dnn/halide_scheduler_opencl_alexnet.yml" :
@@ -106,6 +107,7 @@ TEST_P(DNNTestNetwork, AlexNet)
106107

107108
TEST_P(DNNTestNetwork, ResNet_50)
108109
{
110+
applyTestTag(target == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_512MB : CV_TEST_TAG_MEMORY_1GB);
109111
processNet("dnn/ResNet-50-model.caffemodel", "dnn/ResNet-50-deploy.prototxt",
110112
Size(224, 224), "prob",
111113
target == DNN_TARGET_OPENCL ? "dnn/halide_scheduler_opencl_resnet_50.yml" :
@@ -122,12 +124,14 @@ TEST_P(DNNTestNetwork, SqueezeNet_v1_1)
122124

123125
TEST_P(DNNTestNetwork, GoogLeNet)
124126
{
127+
applyTestTag(target == DNN_TARGET_CPU ? "" : CV_TEST_TAG_MEMORY_512MB);
125128
processNet("dnn/bvlc_googlenet.caffemodel", "dnn/bvlc_googlenet.prototxt",
126129
Size(224, 224), "prob");
127130
}
128131

129132
TEST_P(DNNTestNetwork, Inception_5h)
130133
{
134+
applyTestTag(CV_TEST_TAG_MEMORY_512MB);
131135
double l1 = default_l1, lInf = default_lInf;
132136
if (backend == DNN_BACKEND_INFERENCE_ENGINE && (target == DNN_TARGET_CPU || target == DNN_TARGET_OPENCL))
133137
{
@@ -142,6 +146,7 @@ TEST_P(DNNTestNetwork, Inception_5h)
142146

143147
TEST_P(DNNTestNetwork, ENet)
144148
{
149+
applyTestTag(target == DNN_TARGET_CPU ? "" : CV_TEST_TAG_MEMORY_512MB);
145150
if ((backend == DNN_BACKEND_INFERENCE_ENGINE) ||
146151
(backend == DNN_BACKEND_OPENCV && target == DNN_TARGET_OPENCL_FP16))
147152
throw SkipTestException("");
@@ -153,6 +158,7 @@ TEST_P(DNNTestNetwork, ENet)
153158

154159
TEST_P(DNNTestNetwork, MobileNet_SSD_Caffe)
155160
{
161+
applyTestTag(CV_TEST_TAG_MEMORY_512MB);
156162
if (backend == DNN_BACKEND_HALIDE)
157163
throw SkipTestException("");
158164
Mat sample = imread(findDataFile("dnn/street.png", false));
@@ -184,6 +190,7 @@ TEST_P(DNNTestNetwork, MobileNet_SSD_Caffe_Different_Width_Height)
184190

185191
TEST_P(DNNTestNetwork, MobileNet_SSD_v1_TensorFlow)
186192
{
193+
applyTestTag(target == DNN_TARGET_CPU ? "" : CV_TEST_TAG_MEMORY_512MB);
187194
if (backend == DNN_BACKEND_HALIDE)
188195
throw SkipTestException("");
189196
Mat sample = imread(findDataFile("dnn/street.png", false));
@@ -214,6 +221,7 @@ TEST_P(DNNTestNetwork, MobileNet_SSD_v1_TensorFlow_Different_Width_Height)
214221

215222
TEST_P(DNNTestNetwork, MobileNet_SSD_v2_TensorFlow)
216223
{
224+
applyTestTag(target == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_512MB : CV_TEST_TAG_MEMORY_1GB);
217225
if (backend == DNN_BACKEND_HALIDE)
218226
throw SkipTestException("");
219227
Mat sample = imread(findDataFile("dnn/street.png", false));
@@ -226,6 +234,8 @@ TEST_P(DNNTestNetwork, MobileNet_SSD_v2_TensorFlow)
226234

227235
TEST_P(DNNTestNetwork, SSD_VGG16)
228236
{
237+
applyTestTag(CV_TEST_TAG_LONG, (target == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_1GB : CV_TEST_TAG_MEMORY_2GB),
238+
CV_TEST_TAG_DEBUG_VERYLONG);
229239
if (backend == DNN_BACKEND_HALIDE && target == DNN_TARGET_CPU)
230240
throw SkipTestException("");
231241
double scoreThreshold = (target == DNN_TARGET_OPENCL_FP16 || target == DNN_TARGET_MYRIAD) ? 0.0325 : 0.0;
@@ -238,6 +248,8 @@ TEST_P(DNNTestNetwork, SSD_VGG16)
238248

239249
TEST_P(DNNTestNetwork, OpenPose_pose_coco)
240250
{
251+
applyTestTag(CV_TEST_TAG_LONG, (target == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_1GB : CV_TEST_TAG_MEMORY_2GB),
252+
CV_TEST_TAG_DEBUG_VERYLONG);
241253
if (backend == DNN_BACKEND_HALIDE)
242254
throw SkipTestException("");
243255
#if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_LE(2018050000)
@@ -254,6 +266,8 @@ TEST_P(DNNTestNetwork, OpenPose_pose_coco)
254266

255267
TEST_P(DNNTestNetwork, OpenPose_pose_mpi)
256268
{
269+
applyTestTag(CV_TEST_TAG_LONG, (target == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_1GB : CV_TEST_TAG_MEMORY_2GB),
270+
CV_TEST_TAG_DEBUG_VERYLONG);
257271
if (backend == DNN_BACKEND_HALIDE)
258272
throw SkipTestException("");
259273
#if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_LE(2018050000)
@@ -270,6 +284,7 @@ TEST_P(DNNTestNetwork, OpenPose_pose_mpi)
270284

271285
TEST_P(DNNTestNetwork, OpenPose_pose_mpi_faster_4_stages)
272286
{
287+
applyTestTag(CV_TEST_TAG_LONG, CV_TEST_TAG_MEMORY_1GB);
273288
if (backend == DNN_BACKEND_HALIDE)
274289
throw SkipTestException("");
275290
#if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_LE(2018050000)
@@ -318,6 +333,7 @@ TEST_P(DNNTestNetwork, opencv_face_detector)
318333

319334
TEST_P(DNNTestNetwork, Inception_v2_SSD_TensorFlow)
320335
{
336+
applyTestTag(target == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_512MB : CV_TEST_TAG_MEMORY_1GB);
321337
#if defined(INF_ENGINE_RELEASE)
322338
if (backend == DNN_BACKEND_INFERENCE_ENGINE && target == DNN_TARGET_MYRIAD
323339
&& getInferenceEngineVPUType() == CV_DNN_INFERENCE_ENGINE_VPU_TYPE_MYRIAD_X)
@@ -335,6 +351,7 @@ TEST_P(DNNTestNetwork, Inception_v2_SSD_TensorFlow)
335351

336352
TEST_P(DNNTestNetwork, DenseNet_121)
337353
{
354+
applyTestTag(CV_TEST_TAG_MEMORY_512MB);
338355
if (backend == DNN_BACKEND_HALIDE)
339356
throw SkipTestException("");
340357
// Reference output values are in range [-3.807, 4.605]

‎modules/dnn/test/test_caffe_importer.cpp

Lines changed: 13 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -112,6 +112,8 @@ TEST(Test_Caffe, read_googlenet)
112112
typedef testing::TestWithParam<tuple<bool, Target> > Reproducibility_AlexNet;
113113
TEST_P(Reproducibility_AlexNet, Accuracy)
114114
{
115+
Target targetId = get<1>(GetParam());
116+
applyTestTag(targetId == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_512MB : CV_TEST_TAG_MEMORY_1GB);
115117
bool readFromMemory = get<0>(GetParam());
116118
Net net;
117119
{
@@ -132,7 +134,6 @@ TEST_P(Reproducibility_AlexNet, Accuracy)
132134
ASSERT_FALSE(net.empty());
133135
}
134136

135-
int targetId = get<1>(GetParam());
136137
const float l1 = 1e-5;
137138
const float lInf = (targetId == DNN_TARGET_OPENCL_FP16) ? 3e-3 : 1e-4;
138139

@@ -151,9 +152,9 @@ TEST_P(Reproducibility_AlexNet, Accuracy)
151152
INSTANTIATE_TEST_CASE_P(/**/, Reproducibility_AlexNet, Combine(testing::Bool(),
152153
Values(DNN_TARGET_CPU, DNN_TARGET_OPENCL, DNN_TARGET_OPENCL_FP16)));
153154

154-
#if !defined(_WIN32) || defined(_WIN64)
155155
TEST(Reproducibility_FCN, Accuracy)
156156
{
157+
applyTestTag(CV_TEST_TAG_LONG, CV_TEST_TAG_MEMORY_2GB);
157158
Net net;
158159
{
159160
const string proto = findDataFile("dnn/fcn8s-heavy-pascal.prototxt", false);
@@ -179,10 +180,10 @@ TEST(Reproducibility_FCN, Accuracy)
179180

180181
normAssert(ref, out);
181182
}
182-
#endif
183183

184184
TEST(Reproducibility_SSD, Accuracy)
185185
{
186+
applyTestTag(CV_TEST_TAG_MEMORY_512MB);
186187
Net net;
187188
{
188189
const string proto = findDataFile("dnn/ssd_vgg16.prototxt", false);
@@ -264,10 +265,11 @@ INSTANTIATE_TEST_CASE_P(/**/, Reproducibility_MobileNet_SSD,
264265
typedef testing::TestWithParam<Target> Reproducibility_ResNet50;
265266
TEST_P(Reproducibility_ResNet50, Accuracy)
266267
{
268+
Target targetId = GetParam();
269+
applyTestTag(targetId == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_512MB : CV_TEST_TAG_MEMORY_1GB);
267270
Net net = readNetFromCaffe(findDataFile("dnn/ResNet-50-deploy.prototxt", false),
268271
findDataFile("dnn/ResNet-50-model.caffemodel", false));
269272

270-
int targetId = GetParam();
271273
net.setPreferableBackend(DNN_BACKEND_OPENCV);
272274
net.setPreferableTarget(targetId);
273275

@@ -330,6 +332,7 @@ INSTANTIATE_TEST_CASE_P(/**/, Reproducibility_SqueezeNet_v1_1,
330332

331333
TEST(Reproducibility_AlexNet_fp16, Accuracy)
332334
{
335+
applyTestTag(CV_TEST_TAG_MEMORY_512MB);
333336
const float l1 = 1e-5;
334337
const float lInf = 3e-3;
335338

@@ -375,6 +378,7 @@ TEST(Reproducibility_GoogLeNet_fp16, Accuracy)
375378
// https://github.com/richzhang/colorization
376379
TEST_P(Test_Caffe_nets, Colorization)
377380
{
381+
applyTestTag(target == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_512MB : CV_TEST_TAG_MEMORY_1GB);
378382
checkBackend();
379383

380384
Mat inp = blobFromNPY(_tf("colorization_inp.npy"));
@@ -405,6 +409,7 @@ TEST_P(Test_Caffe_nets, Colorization)
405409

406410
TEST_P(Test_Caffe_nets, DenseNet_121)
407411
{
412+
applyTestTag(CV_TEST_TAG_MEMORY_512MB);
408413
checkBackend();
409414
const string proto = findDataFile("dnn/DenseNet_121.prototxt", false);
410415
const string model = findDataFile("dnn/DenseNet_121.caffemodel", false);
@@ -520,6 +525,8 @@ INSTANTIATE_TEST_CASE_P(Test_Caffe, opencv_face_detector,
520525

521526
TEST_P(Test_Caffe_nets, FasterRCNN_vgg16)
522527
{
528+
applyTestTag(CV_TEST_TAG_LONG, (target == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_1GB : CV_TEST_TAG_MEMORY_2GB));
529+
523530
#if defined(INF_ENGINE_RELEASE)
524531
if (backend == DNN_BACKEND_INFERENCE_ENGINE && (target == DNN_TARGET_OPENCL || target == DNN_TARGET_OPENCL_FP16))
525532
throw SkipTestException("Test is disabled for DLIE OpenCL targets"); // very slow
@@ -536,6 +543,7 @@ TEST_P(Test_Caffe_nets, FasterRCNN_vgg16)
536543

537544
TEST_P(Test_Caffe_nets, FasterRCNN_zf)
538545
{
546+
applyTestTag(target == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_512MB : CV_TEST_TAG_MEMORY_1GB);
539547
if ((backend == DNN_BACKEND_INFERENCE_ENGINE && target == DNN_TARGET_OPENCL_FP16) ||
540548
(backend == DNN_BACKEND_INFERENCE_ENGINE && target == DNN_TARGET_MYRIAD))
541549
throw SkipTestException("");
@@ -547,6 +555,7 @@ TEST_P(Test_Caffe_nets, FasterRCNN_zf)
547555

548556
TEST_P(Test_Caffe_nets, RFCN)
549557
{
558+
applyTestTag(CV_TEST_TAG_LONG, (target == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_512MB : CV_TEST_TAG_MEMORY_2GB));
550559
if ((backend == DNN_BACKEND_INFERENCE_ENGINE && target == DNN_TARGET_OPENCL_FP16) ||
551560
(backend == DNN_BACKEND_INFERENCE_ENGINE && target == DNN_TARGET_MYRIAD))
552561
throw SkipTestException("");

‎modules/dnn/test/test_darknet_importer.cpp

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,7 @@ TEST(Test_Darknet, read_yolo_voc)
7878

7979
TEST(Test_Darknet, read_yolo_voc_stream)
8080
{
81+
applyTestTag(CV_TEST_TAG_MEMORY_1GB);
8182
Mat ref;
8283
Mat sample = imread(_tf("dog416.png"));
8384
Mat inp = blobFromImage(sample, 1.0/255, Size(416, 416), Scalar(), true, false);
@@ -267,6 +268,8 @@ class Test_Darknet_nets : public DNNTestLayer
267268

268269
TEST_P(Test_Darknet_nets, YoloVoc)
269270
{
271+
applyTestTag(CV_TEST_TAG_LONG, CV_TEST_TAG_MEMORY_1GB);
272+
270273
#if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_GE(2019010000)
271274
if (backend == DNN_BACKEND_INFERENCE_ENGINE && target == DNN_TARGET_OPENCL_FP16)
272275
throw SkipTestException("Test is disabled");
@@ -305,6 +308,8 @@ TEST_P(Test_Darknet_nets, YoloVoc)
305308

306309
TEST_P(Test_Darknet_nets, TinyYoloVoc)
307310
{
311+
applyTestTag(CV_TEST_TAG_MEMORY_512MB);
312+
308313
#if defined(INF_ENGINE_RELEASE)
309314
if (backend == DNN_BACKEND_INFERENCE_ENGINE && target == DNN_TARGET_MYRIAD
310315
&& getInferenceEngineVPUType() == CV_DNN_INFERENCE_ENGINE_VPU_TYPE_MYRIAD_X)
@@ -339,6 +344,8 @@ TEST_P(Test_Darknet_nets, TinyYoloVoc)
339344

340345
TEST_P(Test_Darknet_nets, YOLOv3)
341346
{
347+
applyTestTag(CV_TEST_TAG_LONG, (target == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_1GB : CV_TEST_TAG_MEMORY_2GB));
348+
342349
#if defined(INF_ENGINE_RELEASE)
343350
if (backend == DNN_BACKEND_INFERENCE_ENGINE && target == DNN_TARGET_MYRIAD
344351
&& getInferenceEngineVPUType() == CV_DNN_INFERENCE_ENGINE_VPU_TYPE_MYRIAD_X)

‎modules/dnn/test/test_onnx_importer.cpp

Lines changed: 21 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -217,6 +217,7 @@ INSTANTIATE_TEST_CASE_P(/*nothing*/, Test_ONNX_layers, dnnBackendsAndTargets());
217217
class Test_ONNX_nets : public Test_ONNX_layers {};
218218
TEST_P(Test_ONNX_nets, Alexnet)
219219
{
220+
applyTestTag(target == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_512MB : CV_TEST_TAG_MEMORY_1GB);
220221
const String model = _tf("models/alexnet.onnx");
221222

222223
Net net = readNetFromONNX(model);
@@ -270,55 +271,61 @@ TEST_P(Test_ONNX_nets, Googlenet)
270271

271272
TEST_P(Test_ONNX_nets, CaffeNet)
272273
{
274+
applyTestTag(target == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_512MB : CV_TEST_TAG_MEMORY_1GB);
273275
testONNXModels("caffenet", pb);
274276
}
275277

276278
TEST_P(Test_ONNX_nets, RCNN_ILSVRC13)
277279
{
280+
applyTestTag(target == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_512MB : CV_TEST_TAG_MEMORY_1GB);
281+
278282
// Reference output values are in range [-4.992, -1.161]
279283
testONNXModels("rcnn_ilsvrc13", pb, 0.0045);
280284
}
281285

282-
#ifdef OPENCV_32BIT_CONFIGURATION
283-
TEST_P(Test_ONNX_nets, DISABLED_VGG16) // memory usage >2Gb
284-
#else
285286
TEST_P(Test_ONNX_nets, VGG16)
286-
#endif
287287
{
288+
applyTestTag(CV_TEST_TAG_MEMORY_6GB); // > 2.3Gb
289+
288290
// output range: [-69; 72], after Softmax [0; 0.96]
289291
testONNXModels("vgg16", pb, default_l1, default_lInf, true);
290292
}
291293

292-
#ifdef OPENCV_32BIT_CONFIGURATION
293-
TEST_P(Test_ONNX_nets, DISABLED_VGG16_bn) // memory usage >2Gb
294-
#else
295294
TEST_P(Test_ONNX_nets, VGG16_bn)
296-
#endif
297295
{
296+
applyTestTag(CV_TEST_TAG_MEMORY_6GB); // > 2.3Gb
297+
298298
// output range: [-16; 27], after Softmax [0; 0.67]
299299
const double lInf = (target == DNN_TARGET_MYRIAD) ? 0.038 : default_lInf;
300300
testONNXModels("vgg16-bn", pb, default_l1, lInf, true);
301301
}
302302

303303
TEST_P(Test_ONNX_nets, ZFNet)
304304
{
305+
applyTestTag(target == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_512MB : CV_TEST_TAG_MEMORY_1GB);
305306
testONNXModels("zfnet512", pb);
306307
}
307308

308309
TEST_P(Test_ONNX_nets, ResNet18v1)
309310
{
311+
applyTestTag(CV_TEST_TAG_MEMORY_512MB);
312+
310313
// output range: [-16; 22], after Softmax [0, 0.51]
311314
testONNXModels("resnet18v1", pb, default_l1, default_lInf, true);
312315
}
313316

314317
TEST_P(Test_ONNX_nets, ResNet50v1)
315318
{
319+
applyTestTag(CV_TEST_TAG_MEMORY_512MB);
320+
316321
// output range: [-67; 75], after Softmax [0, 0.98]
317322
testONNXModels("resnet50v1", pb, default_l1, default_lInf, true);
318323
}
319324

320325
TEST_P(Test_ONNX_nets, ResNet101_DUC_HDC)
321326
{
327+
applyTestTag(CV_TEST_TAG_VERYLONG);
328+
322329
#if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_GE(2019010000)
323330
if (backend == DNN_BACKEND_INFERENCE_ENGINE)
324331
throw SkipTestException("Test is disabled for DLIE targets");
@@ -334,6 +341,8 @@ TEST_P(Test_ONNX_nets, ResNet101_DUC_HDC)
334341

335342
TEST_P(Test_ONNX_nets, TinyYolov2)
336343
{
344+
applyTestTag(CV_TEST_TAG_MEMORY_512MB);
345+
337346
if (cvtest::skipUnstableTests)
338347
throw SkipTestException("Skip unstable test");
339348
#if defined(INF_ENGINE_RELEASE)
@@ -347,6 +356,7 @@ TEST_P(Test_ONNX_nets, TinyYolov2)
347356
)
348357
throw SkipTestException("Test is disabled for MyriadX");
349358
#endif
359+
350360
// output range: [-11; 8]
351361
double l1 = (target == DNN_TARGET_OPENCL_FP16 || target == DNN_TARGET_MYRIAD) ? 0.017 : default_l1;
352362
double lInf = (target == DNN_TARGET_OPENCL_FP16 || target == DNN_TARGET_MYRIAD) ? 0.14 : default_lInf;
@@ -367,6 +377,7 @@ TEST_P(Test_ONNX_nets, MobileNet_v2)
367377

368378
TEST_P(Test_ONNX_nets, LResNet100E_IR)
369379
{
380+
applyTestTag(target == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_512MB : CV_TEST_TAG_MEMORY_1GB);
370381
if (backend == DNN_BACKEND_INFERENCE_ENGINE &&
371382
(target == DNN_TARGET_OPENCL_FP16 || target == DNN_TARGET_OPENCL || target == DNN_TARGET_MYRIAD))
372383
throw SkipTestException("");
@@ -419,6 +430,8 @@ TEST_P(Test_ONNX_nets, Inception_v2)
419430

420431
TEST_P(Test_ONNX_nets, DenseNet121)
421432
{
433+
applyTestTag(CV_TEST_TAG_MEMORY_512MB);
434+
422435
// output range: [-87; 138], after Softmax [0; 1]
423436
testONNXModels("densenet121", pb, default_l1, default_lInf, true);
424437
}

‎modules/dnn/test/test_tf_importer.cpp

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -361,6 +361,8 @@ TEST_P(Test_TensorFlow_nets, MobileNet_SSD)
361361

362362
TEST_P(Test_TensorFlow_nets, Inception_v2_SSD)
363363
{
364+
applyTestTag(target == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_512MB : CV_TEST_TAG_MEMORY_1GB);
365+
364366
#if defined(INF_ENGINE_RELEASE)
365367
if (backend == DNN_BACKEND_INFERENCE_ENGINE && target == DNN_TARGET_MYRIAD
366368
&& getInferenceEngineVPUType() == CV_DNN_INFERENCE_ENGINE_VPU_TYPE_MYRIAD_X
@@ -426,6 +428,7 @@ TEST_P(Test_TensorFlow_nets, MobileNet_v1_SSD)
426428

427429
TEST_P(Test_TensorFlow_nets, Faster_RCNN)
428430
{
431+
applyTestTag(CV_TEST_TAG_LONG, (target == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_1GB : CV_TEST_TAG_MEMORY_2GB)); // FIXIT split test
429432
static std::string names[] = {"faster_rcnn_inception_v2_coco_2018_01_28",
430433
"faster_rcnn_resnet50_coco_2018_01_28"};
431434

@@ -521,6 +524,8 @@ TEST_P(Test_TensorFlow_nets, opencv_face_detector_uint8)
521524
// np.save('east_text_detection.geometry.npy', geometry)
522525
TEST_P(Test_TensorFlow_nets, EAST_text_detection)
523526
{
527+
applyTestTag(target == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_512MB : CV_TEST_TAG_MEMORY_1GB);
528+
524529
#if defined(INF_ENGINE_RELEASE)
525530
if (backend == DNN_BACKEND_INFERENCE_ENGINE && target == DNN_TARGET_MYRIAD)
526531
throw SkipTestException("Test is disabled for Myriad targets");
@@ -695,6 +700,7 @@ TEST(Test_TensorFlow, two_inputs)
695700

696701
TEST(Test_TensorFlow, Mask_RCNN)
697702
{
703+
applyTestTag(CV_TEST_TAG_MEMORY_1GB);
698704
std::string proto = findDataFile("dnn/mask_rcnn_inception_v2_coco_2018_01_28.pbtxt", false);
699705
std::string model = findDataFile("dnn/mask_rcnn_inception_v2_coco_2018_01_28.pb", false);
700706

0 commit comments

Comments
 (0)