Merge pull request #14284 from dkurt:fix_14236

6 years ago · 5f500692f1
parent 227ec19929 7ce2397b62
commit 5f500692f1
1 changed files with 2 additions and 2 deletions
--- a/modules/dnn/src/tensorflow/tf_importer.cpp
+++ b/modules/dnn/src/tensorflow/tf_importer.cpp
@ -1509,8 +1509,8 @@ void TFImporter::populateNet(Net dstNet)
                if (layerParams.blobs.size() == 2)
                    CV_Error(Error::StsNotImplemented, "Cannot determine number "
                             "of parameters for batch normalization layer.");
-                mean = Mat::zeros(1, layerParams.blobs[3].total(), CV_32F);
-                std = Mat::ones(1, layerParams.blobs[3].total(), CV_32F);
+                mean = Mat::zeros(1, layerParams.blobs[2].total(), CV_32F);
+                std = Mat::ones(1, layerParams.blobs[2].total(), CV_32F);

                // Add an extra layer: Mean-Variance normalization
                LayerParams mvnParams;