mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2024-11-25 19:50:38 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Output blobs shapes initialization in case of identity permutation (NCHW->NCHW)

Browse Source
This commit is contained in:
Dmitry Kurtaev 2017-10-17 17:13:33 +03:00
parent 110af09bf9
commit a3a446c197
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
modules/dnn/src/layers/permute_layer.cpp
View File

@ -117,7 +117,10 @@ public:
std::vector<MatShape> &internals) const
{
if(!_needsPermute)
{
Layer::getMemoryShapes(inputs, requiredOutputs, outputs, internals);
return true;
}
CV_Assert(inputs.size() > 0);
CV_Assert((int)_numAxes == inputs[0].size());
@ -253,7 +256,11 @@ public:
if(!_needsPermute)
{
for (k = 0; k < ninputs; k++)
outputs[k] = *inputs[k];
{
CV_Assert(outputs[k].total() == inputs[k]->total());
if (outputs[k].data != inputs[k]->data)
inputs[k]->copyTo(outputs[k]);
}
}
else
{