fanwuyang/neural-enhance-super-resolution
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Improve instructions to train custom models so a new file is output and existing one is not loaded. Use --model parameter!

Browse Source
This commit is contained in:
Alex J. Champandard
2016-10-30 10:50:57 +01:00
parent 203917d122
commit f68f04fb1c
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.rst
+5 -2
View File
@@ -58,16 +58,19 @@ Pre-trained models are provided in the GitHub releases. Training your own is a
rm -f ne4x*.pkl.bz2
# Pre-train the model using perceptual loss from paper [1] below.
python3.4 enhance.py --train "data/*.jpg" --scales=2 --epochs=50 \
python3.4 enhance.py --train "data/*.jpg" --model custom --scales=2 --epochs=50 \
--perceptual-layer=conv2_2 --smoothness-weight=1e7 --adversary-weight=0.0 \
--generator-blocks=4 --generator-filters=64
# Train the model using an adversarial setup based on [4] below.
python3.4 enhance.py --train "data/*.jpg" --scales=2 --epochs=250 \
python3.4 enhance.py --train "data/*.jpg" --model custom --scales=2 --epochs=250 \
--perceptual-layer=conv5_2 --smoothness-weight=2e4 --adversary-weight=2e5 \
--generator-start=5 --discriminator-start=0 --adversarial-start=5 \
--discriminator-size=64
# The newly trained model is output into this file...
ls ne4x-custom-*.pkl.bz2
.. image:: docs/BankLobby_example.gif