From bb94730fb04d89faeb5575b79ffab3dfdfef488d Mon Sep 17 00:00:00 2001
From: Tiago Freitas Pereira <tiagofrepereira@gmail.com>
Date: Tue, 8 Nov 2016 10:49:42 +0100
Subject: [PATCH] Fixed bug normalizer

---
 bob/learn/tensorflow/datashuffler/Normalizer.py | 3 +--
 bob/learn/tensorflow/network/VGG16_mod.py       | 6 +++---
 2 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/bob/learn/tensorflow/datashuffler/Normalizer.py b/bob/learn/tensorflow/datashuffler/Normalizer.py
index ed65dad1..74e264ef 100644
--- a/bob/learn/tensorflow/datashuffler/Normalizer.py
+++ b/bob/learn/tensorflow/datashuffler/Normalizer.py
@@ -25,9 +25,8 @@ class MeanOffset(object):
         self.mean_offset = mean_offset
 
     def __call__(self, x):
-
         for i in range(len(self.mean_offset)):
-            x[:, i, :, :] = x[:, i, :, :] - self.mean_offset[i]
+            x[:, :, i] = x[:, :, i] - self.mean_offset[i]
 
         return x
 
diff --git a/bob/learn/tensorflow/network/VGG16_mod.py b/bob/learn/tensorflow/network/VGG16_mod.py
index d2e7afa8..d902dc1b 100644
--- a/bob/learn/tensorflow/network/VGG16_mod.py
+++ b/bob/learn/tensorflow/network/VGG16_mod.py
@@ -53,13 +53,13 @@ class VGG16_mod(SequenceNetwork):
 
                  # Forth convolutional block
                  conv5_1_kernel_size=3,
-                 conv5_1_output=512,
+                 conv5_1_output=256,
 
                  conv5_2_kernel_size=3,
-                 conv5_2_output=512,
+                 conv5_2_output=256,
 
                  conv5_3_kernel_size=3,
-                 conv5_3_output=512,
+                 conv5_3_output=256,
 
                  n_classes=10,
 
-- 
GitLab