Administrator/opencv-MIRROR
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2026-09-12 05:11:04 -05:00
Code Issues Packages Projects Releases Wiki Activity

Fix TensorFlow split layer

Browse Source
This commit is contained in:
Dmitry Kurtaev
2017-10-02 22:44:42 +03:00
parent 2a2537725a
commit 2a21c10837
3 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modules/dnn/src/layers/slice_layer.cpp
View File

@@ -116,7 +116,7 @@ public:
}
else // Divide input blob on equal parts by axis.
{
CV_Assert(0 < axis && axis < inpShape.size());
CV_Assert(0 <= axis && axis < inpShape.size());
CV_Assert(requiredOutputs > 0 && inpShape[axis] % requiredOutputs == 0);
inpShape[axis] /= requiredOutputs;
outputs.resize(requiredOutputs, inpShape);