This repository was archived by the owner on Jan 2, 2021. It is now read-only.
File tree Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Original file line number Diff line number Diff line change @@ -58,16 +58,19 @@ Pre-trained models are provided in the GitHub releases. Training your own is a
5858 rm -f ne4x* .pkl.bz2
5959
6060 # Pre-train the model using perceptual loss from paper [1] below.
61- python3.4 enhance.py --train " data/*.jpg" --scales=2 --epochs=50 \
61+ python3.4 enhance.py --train " data/*.jpg" --model custom -- scales=2 --epochs=50 \
6262 --perceptual-layer=conv2_2 --smoothness-weight=1e7 --adversary-weight=0.0 \
6363 --generator-blocks=4 --generator-filters=64
6464
6565 # Train the model using an adversarial setup based on [4] below.
66- python3.4 enhance.py --train " data/*.jpg" --scales=2 --epochs=250 \
66+ python3.4 enhance.py --train " data/*.jpg" --model custom -- scales=2 --epochs=250 \
6767 --perceptual-layer=conv5_2 --smoothness-weight=2e4 --adversary-weight=2e5 \
6868 --generator-start=5 --discriminator-start=0 --adversarial-start=5 \
6969 --discriminator-size=64
7070
71+ # The newly trained model is output into this file...
72+ ls ne4x-custom-* .pkl.bz2
73+
7174
7275 .. image :: docs/BankLobby_example.gif
7376
You can’t perform that action at this time.
0 commit comments