From 521d18ea11cd95d89e90fa002fab3a3efd02ad5d Mon Sep 17 00:00:00 2001
From: Tiago Freitas Pereira <tiagofrepereira@gmail.com>
Date: Fri, 13 Jul 2018 15:04:34 +0200
Subject: [PATCH] Improved the --help

---
 bob/learn/tensorflow/script/style_transfer.py | 61 ++++++++++++++++++-
 1 file changed, 60 insertions(+), 1 deletion(-)

diff --git a/bob/learn/tensorflow/script/style_transfer.py b/bob/learn/tensorflow/script/style_transfer.py
index 50e5c003..1cece002 100644
--- a/bob/learn/tensorflow/script/style_transfer.py
+++ b/bob/learn/tensorflow/script/style_transfer.py
@@ -98,10 +98,69 @@ def style_transfer(content_image_path, output_path, style_image_paths,
                    content_weight, style_weight, denoise_weight, content_end_points,
                    style_end_points, scopes, pure_noise,  **kwargs):
     """
-     Trains neural style transfer from
+     Trains neural style transfer using the approach presented in:
 
     Gatys, Leon A., Alexander S. Ecker, and Matthias Bethge. "A neural algorithm of artistic style." arXiv preprint arXiv:1508.06576 (2015).
 
+    \b
+
+    If you want run a style transfer using InceptionV2 as basis folo
+
+    Below follow a CONFIG template
+    
+    CONFIG.PY
+    ```
+
+       from bob.extension import rc
+
+       from bob.learn.tensorflow.network import inception_resnet_v2_batch_norm
+       architecture = inception_resnet_v2_batch_norm
+
+       checkpoint_dir = rc["bob.bio.face_ongoing.idiap_casia_inception_v2_centerloss_rgb"]
+
+       style_end_points = ["Conv2d_1a_3x3", "Conv2d_2b_3x3", "Conv2d_3b_1x1", "Conv2d_4a_3x3"]
+
+       content_end_points = ["Bottleneck", "PreLogitsFlatten"]
+
+       scopes = {"InceptionResnetV2/":"InceptionResnetV2/"}
+
+    ```
+    \b
+
+    Then run::
+
+       $ bob tf style <content-image> <output-image> --style-image-paths <style-image> CONFIG.py
+
+
+    You can also provide a list of images to encode the style using the config file as in the example below.
+
+    CONFIG.PY
+    ```
+
+       from bob.extension import rc
+
+       from bob.learn.tensorflow.network import inception_resnet_v2_batch_norm
+       architecture = inception_resnet_v2_batch_norm
+
+       checkpoint_dir = rc["bob.bio.face_ongoing.idiap_casia_inception_v2_centerloss_rgb"]
+
+       style_end_points = ["Conv2d_1a_3x3", "Conv2d_2b_3x3", "Conv2d_3b_1x1", "Conv2d_4a_3x3"]
+
+       content_end_points = ["Bottleneck", "PreLogitsFlatten"]
+
+       scopes = {"InceptionResnetV2/":"InceptionResnetV2/"}
+
+       style_image_paths = ["STYLE_1.png",
+                            "STYLE_2.png"]
+
+    ```
+ 
+    Then run::
+
+       $ bob tf style <content-image> <output-image> CONFIG.py
+
+    \b \b
+
     """
 
     # Reading and converting to the tensorflow format
-- 
GitLab