Merge pull request #19674 from l-bat:lb/fix_ie_tests

4 years ago · 2a808aeec0
parent a0008de281 94533e12eb
commit 2a808aeec0
1 changed files with 6 additions and 0 deletions
--- a/modules/dnn/src/ie_ngraph.cpp
+++ b/modules/dnn/src/ie_ngraph.cpp
@ -769,8 +769,14 @@ static InferenceEngine::Layout estimateLayout(const Mat& m)
 {
    if (m.dims == 4)
        return InferenceEngine::Layout::NCHW;
+    else if (m.dims == 3)
+        return InferenceEngine::Layout::CHW;
    else if (m.dims == 2)
        return InferenceEngine::Layout::NC;
+    else if (m.dims == 1)
+        return InferenceEngine::Layout::C;
+    else if (m.dims == 5)
+        return InferenceEngine::Layout::NCDHW;
    else
        return InferenceEngine::Layout::ANY;
 }