1
0
Fork 0
mirror of synced 2024-06-26 18:20:26 +12:00
Commit graph

220 commits

Author SHA1 Message Date
nagadomi 2ffa93213e remove exception handling for cunn 2015-08-05 16:03:51 +09:00
nagadomi dcd43b89fd fix missing nn in latest torch7 2015-08-05 11:49:38 +09:00
nagadomi 54580ba8c0 add data augmentation method that uses overlay 2015-08-02 22:02:14 +09:00
nagadomi 6c65463564 fix slow compression 2015-07-25 02:39:36 +09:00
nagadomi e5a2ef97c7 set compression level of the output file 2015-07-24 17:16:27 +09:00
nagadomi 8e1625d001 change the noisy:cleaned ratio of training data 2015-07-18 11:17:34 +09:00
nagadomi e3d3a8355c change training script
- add AlexNet's color noise (default: false)
- add `photo` category for noise level setting
2015-07-11 21:57:04 +09:00
nagadomi d8ca857a82 change initial weights 2015-07-11 14:52:51 +09:00
nagadomi fd9dadd7a4 use settings.validation_crop 2015-06-26 20:12:51 +09:00
nagadomi 5b4d692f03 add support for RGB color space reconstruction
- add new RGB model (models/anime_style_art_rgb).
- RGB model can reduce color noise.
- waifu2x uses this RGB model by default.

You can use Y model with:
$ th waifu2x.lua -model_dir models/anime_style_art -i input.png -o output.png
$ th train.lua -color y ...
2015-06-23 02:55:30 +09:00
nagadomi 628bd971c9 Merge branch 'master' of github.com:nagadomi/waifu2x into rgb 2015-06-22 23:04:17 +09:00
nagadomi 04c1d7f936 remove unused function 2015-06-22 23:03:14 +09:00
nagadomi 1cdceb1648 throw error when failed to load image 2015-06-22 23:01:56 +09:00
nagadomi bbc3cac273 add support for transparent png (retain the alpha channel) 2015-06-16 20:41:48 +09:00
nagadomi a706892b59 merge develop repo
- remove support for cuDNN
- add new pre-trained model and json files
- some changes in training script

If you have cuDNN model, Please run following commands to convert.
$ cp models/your_own_model.t7 models/your_own_model.t7.backup
$ th cudnn2cunn.lua -model models/your_own_model.t7
2015-06-13 15:20:40 +09:00
nagadomi 1c1f2292d1 fix ugly chroma scaling 2015-05-22 20:06:25 +09:00
nagadomi 0d115f46f6 fix data augmentation in pairwise_transform.scale 2015-05-22 18:59:25 +09:00
nagadomi ffd52fdb2b super typo fix 2015-05-19 16:47:52 +09:00
nagadomi 2231423056 update training script 2015-05-17 14:43:07 +09:00
nagadomi 1273b3609e first commit 2015-05-16 14:48:05 +09:00