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

fix Split partial sum

Browse Source
This commit is contained in:
Smirnov Egor 2021-08-16 15:44:54 +03:00
parent cfb36443fb
commit 9ef41f68fb
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/dnn/src/onnx/onnx_importer.cpp
View File

@ -851,7 +851,7 @@ void ONNXImporter::parseSplit(LayerParams& layerParams, const opencv_onnx::NodeP
std::vector<int> slicePoints(numSplits - 1, splits.get<int>(0)); std::vector<int> slicePoints(numSplits - 1, splits.get<int>(0));
for (int i = 1; i < splits.size() - 1; ++i) 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())); layerParams.set("slice_point", DictValue::arrayInt(&slicePoints[0], slicePoints.size()));
} }

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

@ -617,6 +617,7 @@ TEST_P(Test_ONNX_layers, Split)
testONNXModels("split_2"); testONNXModels("split_2");
testONNXModels("split_3"); testONNXModels("split_3");
testONNXModels("split_4"); testONNXModels("split_4");
testONNXModels("split_sizes");
} }
TEST_P(Test_ONNX_layers, Slice) TEST_P(Test_ONNX_layers, Slice)