Merge pull request #14284 from dkurt:fix_14236

5f500692 · Alexander Alekhin · 227ec199 · 7ce2397b · 5f500692
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

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

未找到文件。
--- 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;