Commit Graph
36 Commits
Author SHA1 Message Date
igv c64c037f01 Local residual learning 2017-10-28 17:07:25 +03:00
igv bd0e4dc965 Support for recursive layers 2017-10-27 14:24:53 +03:00
igv 1a8bb3d9fc Global residual learning 2017-10-24 11:15:09 +03:00
igv cc7d2722d2 Add Charbonnier loss 2018-01-20 16:41:08 +02:00
igv d6b13cce93 Optimizations
Mainly to speed up preprocessing and reduce memory usage
Also get rid of expand_data.py
2017-11-04 13:56:05 +02:00
igv be0ef42af1 Use Wand (ImageMagick) for preprocessing if available instead of Pillow 2017-11-03 20:00:13 +02:00
igv fa94583a94 Support for (MS-)LapSRN 2017-10-21 13:22:25 +03:00
igv e786c07a4b gen.py: generate deconvolution passes
Usable now, except:
    - No support for shrinking/expanding layers
    - Generated shaders for scaling factors other than 2x are not compatible with mpv, they have to be modified manually
2017-10-07 12:52:05 +03:00
igv 9e421c1d33 Half working script for generating a shader
Doesn't support shrinking/expanding layers and doesn't generate a deconvolution passes yet
2017-10-06 15:44:23 +03:00
igv 4a02e34490 Support for ESPCN 2017-10-05 10:30:35 +03:00
igv e0cf791fdf Support for different radiuses for the deconvolution input tensor
Also append radius to the checkpoint directory name.
0.5 1.0
2017-09-10 09:37:39 +03:00
igv 286e6cfc3b Script for sorting deconvolution filter weights 2017-09-08 15:41:49 +03:00
igv 7e828beb9a Make it possible to disable shrinking/expanding layers 2017-09-03 18:42:33 +03:00
igv 5346bcf898 Adjust deconvolution filter size for different scaling factors 0.1 0.2 2017-09-01 14:14:27 +03:00
igv f8af70e011 Append model shape to the checkpoint directory name 2017-08-21 14:06:21 +03:00
igv 15bc798701 Option to add distortions to random low-res images 2017-07-27 14:46:28 +03:00
igv f6b024388f Randomly scale, rotate and flip the training data 2017-07-27 14:40:32 +03:00
igv 82ed39e177 Fixes 2017-07-23 15:32:28 +03:00
Niklas Haas bbbaabe2c0 Switch to the adam optimizer instead of SGD+momentum
Converges faster, and doesn't get stuck forever like momentum
optimization does
2017-07-11 03:00:49 +02:00
Niklas Haas fad61f249b Use MS-SSIM instead of MSE (PSNR) 2017-07-11 02:59:34 +02:00
Niklas Haas db2c179879 utils.py: dump weights in a more machine parsable way
This dumps them as literal Haskell source code, great for writing a GLSL
generator in Haskell - but it should also be valid source code in many
other languages
2017-07-03 00:26:32 +02:00
Niklas Haas d5bf29d6de model.py: also dump alpha variables
In addition, I've also fixed a bug where the last alpha was hard coded
to alpha7, instead of being dynamic relative to m. This would have
introduced a conflict when m was high enough.
2017-07-02 23:29:28 +02:00
Drake Levy e66845dbe0 Misc changes + make code more pythonic 2017-03-14 13:51:11 -07:00
Drake Levy 3368977b38 Fixed README spacing 2017-03-14 13:20:32 -07:00
Drake Levy 322e465ae1 README spacing 2017-03-14 13:19:38 -07:00
Drake Levy 4dfb7a8e71 Modify README links 2017-03-14 13:16:25 -07:00
Drake Levy 8a1941bad3 Added super-resolved image 2017-03-01 18:01:34 -08:00
Drake Levy 007e421178 Almost ready 2017-03-01 17:59:31 -08:00
Drake Levy 906cfe03ff README usage formatting 2017-03-01 17:40:48 -08:00
Drake Levy 9dc08303cc README image formatting 2017-03-01 17:38:49 -08:00
Drake Levy 8562f557b3 Added result images, README 2017-03-01 17:38:07 -08:00
Drake Levy 9fb554e820 README and is_train -> train 2017-03-01 17:29:13 -08:00
Drake Levy 6bdc5f5a29 README usage 2017-03-01 17:27:12 -08:00
Drake Levy bd599621f8 README update 2017-03-01 17:25:03 -08:00
Drake Levy da3ba8f92b Updated README and pretty much everything 2017-03-01 17:23:13 -08:00
Drake Levy 9bf8cf669a Initial commit 2017-03-01 17:16:44 -08:00