From 1f874028fb81114fe8ed27fe813614ae5810ac0b Mon Sep 17 00:00:00 2001 From: Abduragim Shtanchaev Date: Wed, 22 May 2024 08:54:28 +0300 Subject: [PATCH] fixed for openvino einsum --- modules/dnn/src/layers/einsum_layer.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modules/dnn/src/layers/einsum_layer.cpp b/modules/dnn/src/layers/einsum_layer.cpp index 126ce7b5c6..f48eb329ab 100644 --- a/modules/dnn/src/layers/einsum_layer.cpp +++ b/modules/dnn/src/layers/einsum_layer.cpp @@ -498,7 +498,7 @@ public: } else { // Check if there is a pre-processed version of this input // If so assign it to result - if (!preProcessedInputs[0].empty()) + if (!preProcessedInputs.empty() && !preProcessedInputs[0].empty()) { result = preProcessedInputs[0]; } @@ -541,7 +541,7 @@ public: // Use either the preprocessed inputs (if it is available) or the corresponding raw inputs result = pairwiseOperandProcess(!result.empty() ? result : rawInputs[0], !result.empty() ? tmpResult : homogenizedInputDims[0], - !preProcessedInputs[input].empty() ? preProcessedInputs[input] : rawInputs[input], + (!preProcessedInputs.empty() && !preProcessedInputs[input].empty()) ? preProcessedInputs[input] : rawInputs[input], homogenizedInputDims[input], reducedDims, isFinalPair);