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

37 commits

Author SHA1 Message Date
nagadomi edac608f18 Add support for user specified pairwise data for universal filter 2016-07-05 02:42:40 +09:00
nagadomi 307ae40883 Add noise_scale training 2016-06-08 06:39:36 +09:00
nagadomi 51ae485cd1 Add new models
upconv_7 is 2.3x faster than previous model
2016-05-13 09:49:53 +09:00
nagadomi 1407973b88 Use the specified filter 2016-04-12 01:33:17 +09:00
nagadomi 30fe5db735 Add upsampling_filter option 2016-04-02 22:03:27 +09:00
nagadomi 13f702b968 Add support for resizing with gamma correction 2016-03-28 19:07:09 +09:00
nagadomi 8a799e2d56 Add support for noise level 3 2016-03-18 15:47:53 +09:00
nagadomi 4f67f47d3b refactor 2016-03-17 20:03:04 +09:00
nagadomi d4833160c7 Optionalize downsampling filters 2016-03-17 17:58:37 +09:00
nagadomi eea4c31d7b Change parameters 2015-12-04 18:47:33 +09:00
nagadomi 86feb1d4c9 Change the sampling method 2015-12-02 06:53:26 +09:00
nagadomi a975155828 Change downscaling filters when photo training 2015-11-30 17:19:50 +09:00
nagadomi c72ec3112b Add -random_unsharp_mask_rate option for photo 2015-11-27 18:36:36 +09:00
nagadomi d8ba661d6b Add -jpeg_chroma_subsampling_rate option for JPEG denoise training 2015-11-26 17:10:57 +09:00
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