IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[IE][VPU]: Makes ROIAlign to use 2 stage pipeline (to support ma2485) (#2123)

Browse Source
This commit is contained in:
Andrey Sokolov
2020-09-18 18:12:16 +03:00
committed by GitHub
parent 1721a9cc5a
commit 676ce76650
4 changed files with 64 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
inference-engine/tests_deprecated/functional/vpu/common/layers/myriad_layers_roi_align_test.cpp
View File

@@ -11,3 +11,11 @@ INSTANTIATE_TEST_CASE_P(accuracy, myriadLayersTestsROIAlign_smoke,
::testing::ValuesIn(s_ROIAlignNumROIs),
::testing::ValuesIn(s_ROIAlignMode)),
);
INSTANTIATE_TEST_CASE_P(accuracy_faster, myriadLayersTestsROIAlign_smoke,
::testing::Combine(
::testing::ValuesIn(s_ROIAlignLayerInput_Faster),
::testing::ValuesIn(s_ROIAlignLayerParam_Faster),
::testing::ValuesIn(s_ROIAlignNumROIs_Faster),
::testing::ValuesIn(s_ROIAlignMode_Faster)),
);

20
inference-engine/tests_deprecated/functional/vpu/common/layers/myriad_layers_roi_align_test.hpp
View File

@@ -243,16 +243,26 @@ TEST_P(myriadLayersTestsROIAlign_smoke, ROIAlign) {
static std::vector<Dims> s_ROIAlignLayerInput = {
{{5, 256, 160, 157}},
};
static std::vector<roi_align_param> s_ROIAlignLayerParam = {
{{640, 640, 7, 9, 2, 1.4f}},
};
static std::vector<number_rois> s_ROIAlignNumROIs = {
53
};
static std::vector<roi_align_mode> s_ROIAlignMode = {
std::string("avg"),
std::string("max")
std::string("avg"),
std::string("max")
};
static std::vector<Dims> s_ROIAlignLayerInput_Faster = {
{{1, 256, 200, 257}},
};
static std::vector<roi_align_param> s_ROIAlignLayerParam_Faster = {
{{640, 640, 7, 7, 2, 0.25f}},
};
static std::vector<number_rois> s_ROIAlignNumROIs_Faster = {
750
};
static std::vector<roi_align_mode> s_ROIAlignMode_Faster = {
std::string("avg")
};