103 Commits
Author SHA1 Message Date
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 02d2fca6c5 Corrected value for adversarial loss. Don't refactor math the day after stopping coffee. 2016-10-31 20:55:11 +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. ChampandardandGitHub 0c9937a317 Merge pull request #22 from msfeldstein/master
Remove cnmem theano flag.
2016-10-31 13:29:31 +01:00
Alex J. Champandard 064f9dd589 Add three image pre-processing options, improve loading code. 2016-10-31 06:21:52 +01:00
Michael Feldstein fef84c5b44 Remove cnmem theano flag since it doesn't work if you're sharing GPU with display. 2016-10-30 19:24:50 -07: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. ChampandardandGitHub 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. ChampandardandGitHub 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. ChampandardandGitHub 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. ChampandardandGitHub a9b0cd9887 Merge pull request #9 from zuphilip/patch-1
Fix some typos in README.
2016-10-30 17:07:15 +01:00
Alex J. ChampandardandGitHub fcc5e87858 Merge pull request #4 from dribnet/valid_dir
Add valid dir when necessary.
2016-10-30 17:06:33 +01:00
Alex J. ChampandardandGitHub 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 ZumsteinandGitHub 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