From 5bd29a117d6630cc83aa7fcaee0d52272349ef1b Mon Sep 17 00:00:00 2001
From: Tiago Freitas Pereira <tiagofrepereira@gmail.com>
Date: Sun, 23 Oct 2016 18:06:36 +0200
Subject: [PATCH] Reorganizing the trainer

---
 .../tensorflow/script/train_siamese_casia_webface.py   | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/bob/learn/tensorflow/script/train_siamese_casia_webface.py b/bob/learn/tensorflow/script/train_siamese_casia_webface.py
index 26f168a2..5d1f6e9c 100644
--- a/bob/learn/tensorflow/script/train_siamese_casia_webface.py
+++ b/bob/learn/tensorflow/script/train_siamese_casia_webface.py
@@ -49,11 +49,11 @@ def main():
     train_objects = db_casia.objects(groups="world")
     #train_objects = db.objects(groups="world")
     train_labels = [int(o.client_id) for o in train_objects]
-    directory = "/idiap/resource/database/CASIA-WebFace/CASIA-WebFace"
+    directory = "/idiap/temp/tpereira/DEEP_FACE/CASIA_WEBFACE/casia_webface/preprocessed"
 
     train_file_names = [o.make_path(
         directory=directory,
-        extension="")
+        extension=".hdf5")
                         for o in train_objects]
 
     #train_data_shuffler = TripletWithSelectionDisk(train_file_names, train_labels,
@@ -61,12 +61,12 @@ def main():
     #                                               batch_size=BATCH_SIZE)
 
     train_data_shuffler = TripletWithFastSelectionDisk(train_file_names, train_labels,
-                                                       input_shape=[125, 125, 3],
+                                                       input_shape=[224, 224, 3],
                                                        batch_size=BATCH_SIZE)
 
 
     # Preparing train set
-    directory = "/idiap/temp/tpereira/DEEP_FACE/CASIA/preprocessed"
+    directory = "/idiap/temp/tpereira/DEEP_FACE/CASIA_WEBFACE/mobio/preprocessed"
     validation_objects = db_mobio.objects(protocol="male", groups="dev")
     validation_labels = [o.client_id for o in validation_objects]
 
@@ -76,7 +76,7 @@ def main():
                              for o in validation_objects]
 
     validation_data_shuffler = TripletDisk(validation_file_names, validation_labels,
-                                           input_shape=[125, 125, 3],
+                                           input_shape=[224, 224, 3],
                                            batch_size=VALIDATION_BATCH_SIZE)
     # Preparing the architecture
     # LENET PAPER CHOPRA
-- 
GitLab