From 7ce2397b62ea645cf66a77c11a1e4ab452f81660 Mon Sep 17 00:00:00 2001 From: Dmitry Kurtaev Date: Tue, 9 Apr 2019 13:35:34 +0300 Subject: [PATCH] Fix a bug in FusedBatchNorm (TensorFlow) layer importer --- modules/dnn/src/tensorflow/tf_importer.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modules/dnn/src/tensorflow/tf_importer.cpp b/modules/dnn/src/tensorflow/tf_importer.cpp index a1628104f7..547948f6f6 100644 --- 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;