Merge pull request #14951 from alalek:issue_14937

5 years ago · bff90d4684
parent 6fdce865b8 65552bf403
commit bff90d4684
1 changed files with 2 additions and 2 deletions
--- a/modules/dnn/src/layers/convolution_layer.cpp
+++ b/modules/dnn/src/layers/convolution_layer.cpp
@ -460,11 +460,11 @@ public:

        std::vector<Ptr<BackendWrapper> > blobsWrapper;

-        if (newWeightAndBias)
+        if (fusedWeights)
        {
            Mat wm;
            weightsMat.copyTo(wm); // to handle the case of isContinuous() == false
-            wm.reshape(1, blobs[0].dims, blobs[0].size);
+            wm = wm.reshape(1, blobs[0].dims, blobs[0].size);
            blobsWrapper.push_back(Ptr<BackendWrapper>(new VkComBackendWrapper(wm)));
        }
        else