From 440caa7e71e69d7b812042dade9fa7225c04dcc3 Mon Sep 17 00:00:00 2001
From: Tiago Freitas Pereira <tiagofrepereira@gmail.com>
Date: Sun, 14 Feb 2021 17:57:09 +0100
Subject: [PATCH] Changed the default scaler
---
bob/bio/face/tensorflow/preprocessing.py | 11 +++++++----
1 file changed, 7 insertions(+), 4 deletions(-)
diff --git a/bob/bio/face/tensorflow/preprocessing.py b/bob/bio/face/tensorflow/preprocessing.py
index bf87921c..4b92f27f 100644
--- a/bob/bio/face/tensorflow/preprocessing.py
+++ b/bob/bio/face/tensorflow/preprocessing.py
@@ -42,9 +42,10 @@ def get_preprocessor(output_shape):
layers.experimental.preprocessing.RandomFlip("horizontal"),
# FIXED_STANDARDIZATION from https://github.com/davidsandberg/facenet
# [-0.99609375, 0.99609375]
- layers.experimental.preprocessing.Rescaling(
- scale=1 / 128, offset=-127.5 / 128
- ),
+ # layers.experimental.preprocessing.Rescaling(
+ # scale=1 / 128, offset=-127.5 / 128
+ # ),
+ layers.experimental.preprocessing.Rescaling(scale=1 / 255, offset=0),
]
)
return preprocessor
@@ -100,7 +101,9 @@ def prepare_dataset(
ignore_order = tf.data.Options()
ignore_order.experimental_deterministic = False
ds = ds.with_options(ignore_order)
- ds = ds.map(partial(decode_tfrecords, data_shape=data_shape)).prefetch(buffer_size=autotune)
+ ds = ds.map(partial(decode_tfrecords, data_shape=data_shape)).prefetch(
+ buffer_size=autotune
+ )
if shuffle:
ds = ds.shuffle(shuffle_buffer).repeat(epochs)
preprocessor = get_preprocessor(output_shape)
--
GitLab