diff --git a/bob/learn/tensorflow/datashuffler/Normalizer.py b/bob/learn/tensorflow/datashuffler/Normalizer.py index ed65dad1beba9006291e65046d3bd9fff8ed67c1..74e264eff68bbdd3d5c22767d54bb28c07ac8b27 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 d2e7afa89ba02b6e2377da32a65940f60939e12e..d902dc1bd50caa169f42435f871b9463768ef7b4 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,