fix Split partial sum

3 years ago · 9ef41f68fb
parent cfb36443fb
commit 9ef41f68fb
2 changed files with 2 additions and 1 deletions
--- a/modules/dnn/src/onnx/onnx_importer.cpp
+++ b/modules/dnn/src/onnx/onnx_importer.cpp
@ -851,7 +851,7 @@ void ONNXImporter::parseSplit(LayerParams& layerParams, const opencv_onnx::NodeP
        std::vector<int> slicePoints(numSplits - 1, splits.get<int>(0));
        for (int i = 1; i < splits.size() - 1; ++i)
        {
-            slicePoints[i] = slicePoints[i - 1] + splits.get<int>(i - 1);
+            slicePoints[i] = slicePoints[i - 1] + splits.get<int>(i);
        }
        layerParams.set("slice_point", DictValue::arrayInt(&slicePoints[0], slicePoints.size()));
    }
--- a/modules/dnn/test/test_onnx_importer.cpp
+++ b/modules/dnn/test/test_onnx_importer.cpp
@ -617,6 +617,7 @@ TEST_P(Test_ONNX_layers, Split)
    testONNXModels("split_2");
    testONNXModels("split_3");
    testONNXModels("split_4");
+    testONNXModels("split_sizes");
 }

 TEST_P(Test_ONNX_layers, Slice)