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