1
0
Fork 0
mirror of synced 2024-09-21 20:11:38 +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