Determine layout

94533e12 · Liubov Batanina · 0689c70d · 94533e12
隐藏空白更改
内联并排

Showing with 6 addition and 0 deletion

modules/dnn/src/ie_ngraph.cpp modules/dnn/src/ie_ngraph.cpp +6 -0

未找到文件。
--- 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;
 }