Alex J. Champandard
93e5a41d9a
Fix and optimize pre-processing of images.
2016-11-01 20:40:10 +01:00
Alex J. Champandard
11ba505252
Fix for gradient clipping code.
2016-11-01 16:04:33 +01:00
Alex J. Champandard
cf65207a2e
Use full range of tanh output rather than [-0.5, +0.5], avoids clipping.
2016-11-01 16:02:18 +01:00
Alex J. Champandard
c610623b11
Add gradient clipping, helpful for preventing problems with extreme parameters/architectures.
2016-10-31 21:38:23 +01:00
Alex J. Champandard
f2494f8078
Add new downscale layers, separate from upscale steps. Renamed --scales to --zoom for inference.
2016-10-31 13:49:17 +01:00
Alex J. Champandard
064f9dd589
Add three image pre-processing options, improve loading code.
2016-10-31 06:21:52 +01:00
Alex J. Champandard
5ef872b876
Add warning for files that may be too large for 4x.
2016-10-30 20:45:10 +01:00
Alex J. Champandard and GitHub
a5ad2c25e6
Merge pull request #18 from alexjc/training
...
Training Improvements
2016-10-30 20:29:13 +01:00
Alex J. Champandard
17fcad8d28
Refactor of changes related to training.
2016-10-30 20:26:03 +01:00
Alex J. Champandard and GitHub
2b67daedb6
Merge pull request #12 from dribnet/generic_seeds
...
Move generation of seeds out of training network.
2016-10-30 17:19:33 +01:00
Alex J. Champandard and GitHub
cad5eff572
Merge pull request #11 from dribnet/save_every_epoch
...
Added --save-every-epoch option
2016-10-30 17:18:33 +01:00
Alex J. Champandard and GitHub
a9b0cd9887
Merge pull request #9 from zuphilip/patch-1
...
Fix some typos in README.
2016-10-30 17:07:15 +01:00
Alex J. Champandard and GitHub
fcc5e87858
Merge pull request #4 from dribnet/valid_dir
...
Add valid dir when necessary.
2016-10-30 17:06:33 +01:00
Alex J. Champandard and GitHub
1478977f18
Merge pull request #10 from OndraM/fix-duplicate-param
...
Fix duplicate param definition.
2016-10-30 16:53:48 +01:00
Tom White
8f5167d235
Fix enhancer.process to pass img, seed
2016-10-31 02:14:24 +13:00
Tom White
37cb208374
Move generation of seeds out of training network
...
This moves the generation of the image seeds out of the
training network and into the DataLoader. Currently seeds
are computed as bilinear downsamplings of the original image.
This is almost functionly equivalent to the version
it replaces, but opens up new possibilities at training
time because the seeds are now decoupled from the netork.
For example, seeds could be made with different interpolations
or even with other transformations such as image compression.
2016-10-31 01:25:55 +13:00
Tom White
b05ee6ad08
Added --save-every-epoch option
2016-10-31 00:12:28 +13:00
Tom White
c5053806bd
Add valid dir when necessary
2016-10-31 00:05:37 +13:00
Ondřej Machulda
eb25e737cf
Fix duplicate param definition
2016-10-30 11:48:53 +01:00
Philipp Zumstein and GitHub
f83e69e96a
Fix some typos in README
2016-10-30 11:11:54 +01:00
Alex J. Champandard
f68f04fb1c
Improve instructions to train custom models so a new file is output and existing one is not loaded. Use --model parameter!
2016-10-30 10:50:57 +01:00
Alex J. Champandard
203917d122
Switch default to small model to reduce memory usage.
2016-10-30 10:49:23 +01:00
Alex J. Champandard
2b5fc8f51d
Add docker instructions, fix for slow compute in CPU image.
2016-10-30 02:04:28 +02:00
Alex J. Champandard
74cb95609e
Fix for docker build using latest Miniconda and Python 3.5 explicitly.
2016-10-29 23:57:08 +02:00
Alex J. Champandard
99c767b7e2
Add docker configuration files for CPU and GPU.
2016-10-29 22:29:30 +02:00
Alex J. Champandard
bf22450b8d
Update documentation for new --train usage, minor improvements.
2016-10-29 14:15:24 +02:00
Alex J. Champandard
4c55c48f62
Add argument for specifying training images, cleaned up file handling.
2016-10-29 12:39:07 +02:00
Alex J. Champandard
1c38f2ca31
New meme-friendly image and link to demo.
2016-10-29 01:35:44 +02:00
Alex J. Champandard
b1c054ce9f
Improve the README for applying and training models.
2016-10-29 00:39:32 +02:00
Alex J. Champandard
f868514be3
Improve code for simply applying super-resolution.
2016-10-29 00:18:44 +02:00
Alex J. Champandard
30534c6dd1
Add old station example, fix bank example.
2016-10-28 22:55:10 +02:00
Alex J. Champandard
801a4707f4
Improve README and the visual examples.
2016-10-28 16:29:37 +02:00
Alex J. Champandard
39e12ea205
Add example GIF to the README, tweak text.
2016-10-28 14:38:28 +02:00
Alex J. Champandard
c456221cb5
Experiment with recursive super-resolution and weight reuse, mixed results.
2016-10-28 04:28:30 +02:00
Alex J. Champandard
87304c93a6
Use traditional learning rate decaying rather than fast-restarts, works better when training continuously adapting GAN.
2016-10-28 04:27:17 +02:00
Alex J. Champandard
3809e9b02a
Add loading of images into a buffer, using multiple fragments per JPG loaded. Works well with larger datasets like OpenImages, fully GPU bound.
2016-10-28 04:25:10 +02:00
Alex J. Champandard
619fad7f3c
Add loading parameters from saved models. Clean up learning-rate code.
2016-10-11 17:24:10 +02:00
Alex J. Champandard
bb3b24b04c
Add README and requirements file.
2016-10-11 17:16:06 +02:00
Alex J. Champandard
f978f058dd
Add descriptions for command-line arguments.
2016-10-11 11:12:49 +02:00
Alex J. Champandard
d8c9292b80
Tweak code for generalized super-resolution.
2016-10-11 08:59:42 +02:00
Alex J. Champandard
9558a11397
Add command-line for super-resolving images without training.
2016-10-07 21:05:38 +02:00
Alex J. Champandard
03a6813e95
Enable adversarial loss during training.
2016-10-05 14:52:28 +02:00
Alex J. Champandard
6542f435b4
Stabilize the training, tuned the architecture.
2016-10-05 00:27:26 +02:00
Alex J. Champandard
4580e5531a
Add .gitignore and removed symlink.
2016-10-04 13:19:53 +02:00
Alex J. Champandard
b52530cf20
Add support for PRELU (default), optional load/save.
2016-10-04 11:25:03 +02:00
Alex J. Champandard
4d82c7f157
Improve console output.
2016-10-04 02:16:34 +02:00
Alex J. Champandard
8770113f9a
Add support for loading/saving models. Fixed regressions.
2016-10-04 01:40:30 +02:00
Alex J. Champandard
58e1bed4a6
Improve display and code structure.
2016-10-03 22:17:37 +02:00
Alex J. Champandard
749b467f94
Switch to sub-pixel deconvolution layer.
2016-10-02 21:25:02 +02:00
Alex J. Champandard
9f13695050
Add discriminator network too.
2016-10-02 14:01:51 +02:00