1
0
mirror of https://github.com/opencv/opencv.git synced 2026-07-29 23:33:05 +04:00

Fix deconvolution layer. Add batch norm layer with mean-variance normalization from TensorFlow.

This commit is contained in:
Dmitry Kurtaev
2018-02-12 18:55:27 +03:00
parent ab0f0f26a1
commit a6baedd02c
4 changed files with 136 additions and 18 deletions
+8
View File
@@ -273,6 +273,14 @@ public:
int i, newRows = 1;
for( i = 0; i < splitDim; i++ )
newRows *= inpBlob.size[i];
if (inpBlob.total() == newRows)
{
// MVN is applied to single values at an every row.
outBlob.setTo(0);
return;
}
Mat inpMat = inpBlob.reshape(1, newRows);
Mat outMat = outBlob.reshape(1, newRows);