mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-06-13 13:13:26 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update assertions in batch norm layer

Browse Source
This commit is contained in:
Dmitry Kurtaev 2018-03-07 15:57:15 +03:00
parent 69a8f110b6
commit ab20d2a3fc
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
modules/dnn/src/layers/batch_norm_layer.cpp
View File

@ -32,7 +32,7 @@ public:
BatchNormLayerImpl(const LayerParams& params) BatchNormLayerImpl(const LayerParams& params)
{ {
setParamsFrom(params); setParamsFrom(params);
CV_Assert(blobs.size() >= 3); CV_Assert(blobs.size() >= 2);
hasWeights = params.get<bool>("has_weight", false); hasWeights = params.get<bool>("has_weight", false);
hasBias = params.get<bool>("has_bias", false); hasBias = params.get<bool>("has_bias", false);
@ -46,8 +46,8 @@ public:
blobs[0].type() == CV_32F && blobs[1].type() == CV_32F); blobs[0].type() == CV_32F && blobs[1].type() == CV_32F);
float varMeanScale = 1.f; float varMeanScale = 1.f;
if (!hasWeights && !hasBias) { if (!hasWeights && !hasBias && blobs.size() > 2) {
CV_Assert(blobs[2].type() == CV_32F); CV_Assert(blobs.size() == 3, blobs[2].type() == CV_32F);
varMeanScale = blobs[2].at<float>(0); varMeanScale = blobs[2].at<float>(0);
if (varMeanScale != 0) if (varMeanScale != 0)
varMeanScale = 1/varMeanScale; varMeanScale = 1/varMeanScale;