Merge pull request #14284 from dkurt:fix_14236

5f500692 · Alexander Alekhin · 227ec199 · 7ce2397b · 5f500692
Commit 5f500692 authored Apr 12, 2019 by Alexander Alekhin
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

tf_importer.cpp modules/dnn/src/tensorflow/tf_importer.cpp +2 -2

No files found.
--- 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;