potok/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2026-07-29 15:23:05 +04:00
Code Packages Releases Wiki Activity

Torch's Concat and ConcatTable doesn't use Split layer

Browse Source
This commit is contained in:
Dmitry Kurtaev
2017-08-16 21:11:59 +03:00
parent 8ffa29473f
commit 0ce7c33bc8
2 changed files with 9 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/dnn/src/layers/split_layer.cpp
+2 -3
View File
@@ -75,7 +75,7 @@ public:
Layer::getMemoryShapes(inputs, max(1, outputsCount >= 0 ? outputsCount : requiredOutputs),
outputs, internals);
return true;
return false;
}
void forward(std::vector<Mat*> &inputs, std::vector<Mat> &outputs, std::vector<Mat> &internals)
@@ -86,8 +86,7 @@ public:
for (size_t i = 0; i < outputs.size(); i++)
{
CV_Assert(inputs[0]->total() == outputs[i].total());
if (outputs[i].data != inputs[0]->data)
inputs[0]->copyTo(outputs[i]);
inputs[0]->copyTo(outputs[i]);
}
}
};