1
0
Fork 0
mirror of synced 2024-07-03 05:31:08 +12:00
Commit graph

23 commits

Author SHA1 Message Date
nagadomi abedec9d9b Fix filter name for graphicsmagick
graphicsmagick package has bug in the definition of FilteTypes.
I will send PR to fix that bug.
This commit is related to it.
You should upgrade graphicsmagick package.

$ luarocks install graphicsmagick
2015-11-21 11:57:33 +09:00
nagadomi b5db84d42e Change the jpeg config for the photo model 2015-11-15 09:36:40 +09:00
nagadomi b63e08905b Remove unused variable 2015-11-11 10:54:28 +09:00
nagadomi eaeaef10b7 more flip 2015-11-09 23:45:01 +09:00
nagadomi 15f3f2d901 typo 2015-11-07 07:38:45 +09:00
nagadomi 3ea16b3b86 tunable parameters 2015-11-07 07:18:22 +09:00
nagadomi cf862782a5 depth 8 2015-11-07 06:39:47 +09:00
nagadomi c773e18e59 Add trade-off parameter for noise reduction 2015-11-07 06:37:53 +09:00
nagadomi 903d945652 cleanup 2015-11-06 10:08:54 +09:00
nagadomi b35a9ae7d7 tuning 2015-11-03 06:10:44 +09:00
nagadomi 4c691b4640 refactor 2015-10-29 22:44:15 +09:00
nagadomi da786e15ba remove noise_scale training 2015-10-28 16:27:31 +09:00
nagadomi 3abc5a03e3 refactor 2015-10-28 16:01:07 +09:00
nagadomi 8dea362bed sync from internal repo
- Memory compression by snappy (lua-csnappy)
- Use RGB-wise Weighted MSE(R*0.299, G*0.587, B*0.114) instead of MSE
- Aggressive cropping for edge region
and some change.
2015-10-26 09:23:52 +09:00
nagadomi 54580ba8c0 add data augmentation method that uses overlay 2015-08-02 22:02:14 +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 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 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 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