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