Merge pull request #1166 from dkurt:fix_net_deployment

8 years ago · 74e439f4db
parent dea8d5f928 743f82c002
commit 74e439f4db
1 changed files with 2 additions and 2 deletions
--- a/modules/dnn/src/dnn.cpp
+++ b/modules/dnn/src/dnn.cpp
@ -604,10 +604,10 @@ void Net::setBlob(String outputName, const Mat &blob_)

    LayerData &ld = impl->layers[pin.lid];
    ld.outputBlobs.resize( std::max(pin.oid+1, (int)ld.requiredOutputs.size()) );
-    MatSize prevShape = ld.outputBlobs[pin.oid].size;
+    bool oldShape = ld.outputBlobs[pin.oid].size == blob_.size;
    ld.outputBlobs[pin.oid] = blob_.clone();

-    impl->netWasAllocated = impl->netWasAllocated && prevShape == blob_.size;
+    impl->netWasAllocated = impl->netWasAllocated && oldShape;
 }

 Mat Net::getBlob(String outputName)