From fd5337d5324c260e97ee8abd94303cf641336ecc Mon Sep 17 00:00:00 2001 From: Liubov Batanina Date: Thu, 4 Mar 2021 13:05:01 +0300 Subject: [PATCH] Determine layout original commit: 94533e12ebd6e131723a53e2d31e97986fea6e5b --- modules/dnn/src/ie_ngraph.cpp | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/modules/dnn/src/ie_ngraph.cpp b/modules/dnn/src/ie_ngraph.cpp index c646c1fe3a..49717f8513 100644 --- a/modules/dnn/src/ie_ngraph.cpp +++ b/modules/dnn/src/ie_ngraph.cpp @@ -772,8 +772,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; }