diff --git a/inference-engine/tests/functional/plugin/cpu/single_layer_tests/region_yolo.cpp b/inference-engine/tests/functional/plugin/cpu/single_layer_tests/region_yolo.cpp index 19540f56da65ed..4c88cf8d884873 100644 --- a/inference-engine/tests/functional/plugin/cpu/single_layer_tests/region_yolo.cpp +++ b/inference-engine/tests/functional/plugin/cpu/single_layer_tests/region_yolo.cpp @@ -103,6 +103,8 @@ const std::vector inShapes_mxnet = { {1, 75, 16, 16}, {1, 75, 13, 13}, {1, 75, 8, 8}, + {1, 303, 7, 7}, + {1, 303, 14, 14}, {1, 303, 28, 28}, }; diff --git a/ngraph/test/backend/region_yolo.in.cpp b/ngraph/test/backend/region_yolo.in.cpp index 77292e74e6332d..3172fcb50b83d0 100644 --- a/ngraph/test/backend/region_yolo.in.cpp +++ b/ngraph/test/backend/region_yolo.in.cpp @@ -70,3 +70,33 @@ NGRAPH_TEST(${BACKEND_NAME}, region_yolo_v3_mxnet) shape, TEST_FILES, "region_out_yolov3_mxnet.data"); test_case.run_with_tolerance_as_fp(1.0e-4f); } + +NGRAPH_TEST(${BACKEND_NAME}, region_yolo_v3_mxnet_2) +{ + const size_t num = 3; + const size_t coords = 4; + const size_t classes = 20; + const size_t batch = 1; + const size_t channels = 303; + const size_t width = 28; + const size_t height = 28; + const std::vector mask{0, 1, 2}; + const int axis = 1; + const int end_axis = 3; + + Shape input_shape{batch, channels, height, width}; + Shape output_shape{batch, (classes + coords + 1) * mask.size(), height, width}; + + const auto A = make_shared(element::f32, input_shape); + const auto R = make_shared(A, coords, classes, num, false, mask, axis, end_axis); + const auto f = make_shared(R, ParameterVector{A}); + + EXPECT_EQ(R->get_output_shape(0), output_shape); + + auto test_case = test::TestCase(f); + + test_case.add_input_from_file(input_shape, TEST_FILES, "region_in_yolov3_mxnet_2.data"); + test_case.add_expected_output_from_file( + output_shape, TEST_FILES, "region_out_yolov3_mxnet_2.data"); + test_case.run_with_tolerance_as_fp(1.0e-4f); +} diff --git a/ngraph/test/files/region_in_yolov3_mxnet_2.data b/ngraph/test/files/region_in_yolov3_mxnet_2.data new file mode 100644 index 00000000000000..89bb00a19fe38c Binary files /dev/null and b/ngraph/test/files/region_in_yolov3_mxnet_2.data differ diff --git a/ngraph/test/files/region_out_yolov3_mxnet_2.data b/ngraph/test/files/region_out_yolov3_mxnet_2.data new file mode 100644 index 00000000000000..206bc6d4877094 Binary files /dev/null and b/ngraph/test/files/region_out_yolov3_mxnet_2.data differ