diff --git a/modules/dnn/src/ie_ngraph.cpp b/modules/dnn/src/ie_ngraph.cpp index 84b984ac97a9a9c7550aa1d3bd3bfeddc596ab6e..aa3be70e05ee45b77d12dd134836e950f0846ffa 100644 --- 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; }