mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-06-11 11:45:30 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix 1D handling issue in inner product

Browse Source
This commit is contained in:
Abduragim 2024-01-22 20:10:34 +04:00
parent c739117a7c
commit 0e6b7f1656
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
modules/dnn/src/layers/einsum_layer.cpp
View File

@ -1367,7 +1367,7 @@ Mat LayerEinsumImpl::batchwiseMatMul(
// input1 should of size MxK
// check if input1 needs reshape, if need reshape
if (input1.dims > 2 || input1.size[0] != M || input1.size[1] != K)
if (input1.dims > 2 || input1.size[0] != M || (input1.dims > 1 && input1.size[1] != K) || input1.dims == 1)
{
int shape[] = {M, K};
reshapedInput1 = input1.reshape(1, 2, shape);
@ -1375,7 +1375,7 @@ Mat LayerEinsumImpl::batchwiseMatMul(
// input2 should be of size KxN
// check if input2 needs reshape, if needs reshape
if (input2.dims > 2 || input2.size[0] != K || input2.size[1] != N)
if (input2.dims > 2 || input2.size[0] != K || (input2.dims > 1 && input2.size[1] != N) || input2.dims == 1)
{
int shape2[] = {K, N};
reshapedInput2 = input2.reshape(1, 2, shape2);

2
modules/dnn/test/test_onnx_importer.cpp
View File

@ -1496,7 +1496,7 @@ TEST_P(Test_ONNX_layers, Einsum_5D)
}
// https://github.com/opencv/opencv/issues/24883
TEST_P(Test_ONNX_layers, DISABLED_Einsum_InnerProduct)
TEST_P(Test_ONNX_layers, Einsum_InnerProduct)
{
testONNXModels("einsum_inner", npy, 0, 0, false, false, 2);
}