Update README.md

ahbrosha · Mar 4, 2017 · 5e4c722 · 5e4c722
1 parent e0d3bcc
commit 5e4c722
Show file tree

Hide file tree

Showing 8 changed files with 41 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -58,6 +58,47 @@ style blend weight of 0.2:
 ![input-style](examples/2-style1.jpg)
 ![input-style](examples/2-style2.jpg)
 
+## Tweaking
+
+`--style-layer-weight-exp` command line argument could be used to tweak how "abstract"
+the style transfer should be. Lower values mean that style transfer of a finer features
+will be favored over style transfer of a more coarse features, and vice versa. Default
+value is 1.0 - all layers treated equally. Somewhat extreme examples of what you can achieve:
+
+![--style-layer-weight-exp 0.2](examples/tweaks/swe02.jpg)
+![--style-layer-weight-exp 2.0](examples/tweaks/swe20.jpg)
+
+(**left**: 0.2 - finer features style transfer; **right**: 2.0 - coarser features style trasnfer)
+
+`--content-weight-blend` specifies the coefficient of content transfer layers. Default value -
+1.0, style transfer tries to preserve finer grain content details. The value should be
+in range [0.0; 1.0].
+
+![--content-weight-blend 1.0](examples/tweaks/cwe10_default.jpg)
+![--content-weight-blend 0.1](examples/tweaks/cwe01.jpg)
+
+(**left**: 1.0 - default value; **right**: 0.1 - more abstract picture)
+
+`--pooling` allows to select which pooling layers to use (specify either `max` or `avg`).
+Original VGG topology uses max pooling, but the [style transfer paper][paper] suggests
+replacing it with average pooling. The outputs are perceptually differnt, max pool in
+general tends to have finer detail style trasnfer, but could have troubles at
+lower-freqency detail level:
+
+![--pooling max](examples/tweaks/swe14_pmax.jpg)
+![--pooling avg](examples/tweaks/swe14_pavg.jpg)
+
+(**left**: max pooling; **right**: average pooling)
+
+`--preserve-colors` boolean command line argument adds post-processing step, which
+combines colors from the original image and luma from the stylized image (YCbCr color
+space), thus producing color-preserving style trasnfer:
+
+![--pooling max](examples/tweaks/swe14_pmax.jpg)
+![--pooling max](examples/tweaks/swe14_pmax_pcyuv.jpg)
+
+(**left**: original stylized image; **right**: color-preserving style transfer)
+
 ## Requirements
 
 * [TensorFlow](https://www.tensorflow.org/versions/master/get_started/os_setup.html#download-and-setup)

diff --git a/examples/tweaks/cwe01.jpg b/examples/tweaks/cwe01.jpg
diff --git a/examples/tweaks/cwe10_default.jpg b/examples/tweaks/cwe10_default.jpg
diff --git a/examples/tweaks/swe02.jpg b/examples/tweaks/swe02.jpg
diff --git a/examples/tweaks/swe14_pavg.jpg b/examples/tweaks/swe14_pavg.jpg
diff --git a/examples/tweaks/swe14_pmax.jpg b/examples/tweaks/swe14_pmax.jpg
diff --git a/examples/tweaks/swe14_pmax_pcyuv.jpg b/examples/tweaks/swe14_pmax_pcyuv.jpg
diff --git a/examples/tweaks/swe20.jpg b/examples/tweaks/swe20.jpg