potok/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2026-07-30 07:43:03 +04:00
Code Packages Releases Wiki Activity

nn.BatchNormalization and nn.Dropout layers from Torch

Browse Source
This commit is contained in:
Dmitry Kurtaev
2017-12-04 12:57:21 +03:00
parent fc9e031454
commit bbbec300a6
3 changed files with 22 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/dnn/src/layers/batch_norm_layer.cpp
+3 -2
View File
@@ -119,8 +119,9 @@ public:
CV_Assert(inputs.size() == 1);
Mat &inpBlob = *inputs[0];
int rows = inpBlob.size[2];
int cols = inpBlob.size[3];
CV_Assert(inpBlob.dims == 2 || inpBlob.dims == 4);
int rows = inpBlob.dims > 2 ? inpBlob.size[2] : 1;
int cols = inpBlob.dims > 2 ? inpBlob.size[3] : 1;
for (size_t ii = 0; ii < outputs.size(); ii++)
{