diff --git a/scripts/small-1x.sh b/scripts/small-1x.sh new file mode 100644 index 0000000..1e60c92 --- /dev/null +++ b/scripts/small-1x.sh @@ -0,0 +1,18 @@ +#!/bin/sh + +python3.4 enhance.py \ + --train "$OPEN_IMAGES_PATH/*/*.jpg" --model small \ + --epochs=50 --batch-shape=192 --device=gpu0 \ + --generator-downscale=2 --generator-upscale=2 \ + --generator-blocks=8 --generator-filters=64 \ + --perceptual-layer=conv2_2 --smoothness-weight=1e7 --adversary-weight=0.0 \ + --train-blur=3 --train-noise=5.0 + +python3.4 enhance.py \ + --train "$OPEN_IMAGES_PATH/*/*.jpg" --model small \ + --epochs=500 --batch-shape=192 --device=gpu0 \ + --generator-downscale=2 --generator-upscale=2 \ + --perceptual-layer=conv5_2 --smoothness-weight=2e4 --adversary-weight=2e2 \ + --generator-start=5 --discriminator-start=0 --adversarial-start=5 \ + --discriminator-size=32 \ + --train-blur=3 --train-noise=5.0 diff --git a/scripts/small-2x.sh b/scripts/small-2x.sh new file mode 100644 index 0000000..2e8bdc2 --- /dev/null +++ b/scripts/small-2x.sh @@ -0,0 +1,18 @@ +#!/bin/sh + +python3.4 enhance.py \ + --train "$OPEN_IMAGES_PATH/*/*.jpg" --model small \ + --epochs=50 --batch-shape=192 --device=gpu0 \ + --generator-downscale=1 --generator-upscale=2 \ + --generator-blocks=8 --generator-filters=64 \ + --perceptual-layer=conv2_2 --smoothness-weight=1e7 --adversary-weight=0.0 \ + --train-blur=2 --train-noise=4.0 + +python3.4 enhance.py \ + --train "$OPEN_IMAGES_PATH/*/*.jpg" --model small \ + --epochs=500 --batch-shape=192 --device=gpu0 \ + --generator-downscale=1 --generator-upscale=2 \ + --perceptual-layer=conv5_2 --smoothness-weight=2e4 --adversary-weight=2e2 \ + --generator-start=5 --discriminator-start=0 --adversarial-start=5 \ + --discriminator-size=32 \ + --train-blur=2 --train-noise=4.0 diff --git a/scripts/small-4x.sh b/scripts/small-4x.sh new file mode 100644 index 0000000..17d0493 --- /dev/null +++ b/scripts/small-4x.sh @@ -0,0 +1,18 @@ +#!/bin/sh + +python3.4 enhance.py \ + --train "$OPEN_IMAGES_PATH/*/*.jpg" --model small \ + --epochs=50 --batch-shape=192 --device=gpu0 \ + --generator-downscale=0 --generator-upscale=2 \ + --generator-blocks=8 --generator-filters=64 \ + --perceptual-layer=conv2_2 --smoothness-weight=1e7 --adversary-weight=0.0 \ + --train-blur=1 --train-noise=3.0 + +python3.4 enhance.py \ + --train "$OPEN_IMAGES_PATH/*/*.jpg" --model small \ + --epochs=500 --batch-shape=192 --device=gpu0 \ + --generator-downscale=0 --generator-upscale=2 \ + --perceptual-layer=conv5_2 --smoothness-weight=2e4 --adversary-weight=2e2 \ + --generator-start=5 --discriminator-start=0 --adversarial-start=5 \ + --discriminator-size=32 \ + --train-blur=1 --train-noise=3.0