mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-07-25 22:57:53 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix dnn caffe importer extract blobs from reused layers

Browse Source
This commit is contained in:
Adam Radomski 2018-10-08 17:09:14 +02:00
parent 78a802baae
commit cc3ec5d453
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
modules/dnn/src/caffe/caffe_importer.cpp
View File

@ -278,11 +278,13 @@ public:
int li; int li;
for (li = 0; li != netBinary.layer_size(); li++) for (li = 0; li != netBinary.layer_size(); li++)
{ {
if (netBinary.layer(li).name() == name) const caffe::LayerParameter& binLayer = netBinary.layer(li);
// Break if the layer name is the same and the blobs are not cleared
if (binLayer.name() == name && binLayer.blobs_size() != 0)
break; break;
} }
if (li == netBinary.layer_size() || netBinary.layer(li).blobs_size() == 0) if (li == netBinary.layer_size())
return; return;
caffe::LayerParameter* binLayer = netBinary.mutable_layer(li); caffe::LayerParameter* binLayer = netBinary.mutable_layer(li);