nagadomi
f11108cba1
Add comment about RNG seed
2016-09-14 21:19:59 +09:00
nagadomi
81df729a8a
Remove -loss option
2016-07-09 15:05:11 +09:00
nagadomi
06e073089b
Fix for edac608f
2016-07-06 08:04:24 +09:00
nagadomi
d8fcde46f3
Change default -validation_crops to 200.
2016-06-20 15:56:26 +09:00
nagadomi
f61d59d28a
Change default params. enabling jpeg_chroma_subsampling by default
2016-06-18 20:27:34 +09:00
nagadomi
3b1b2d03fe
Change default -nr_rate to 0.65
2016-06-13 12:44:32 +09:00
nagadomi
b8ff8c6787
Remove -gamma_correction option
2016-06-10 07:37:39 +09:00
nagadomi
01b2e6d441
Remove -upsampling_filter option
2016-06-10 07:34:11 +09:00
nagadomi
e5cfd3dfce
Add -resume option
2016-06-09 02:39:52 +09:00
nagadomi
307ae40883
Add noise_scale training
2016-06-08 06:39:36 +09:00
nagadomi
d0630d3a20
individual filters and box-only support
2016-06-06 14:04:13 +09:00
nagadomi
c89fd7249a
Add learning_rate_decay
2016-06-02 10:11:15 +09:00
nagadomi
8088460a20
Add oracle_rate option
2016-05-27 16:54:29 +09:00
nagadomi
7814691cbf
Add resize_blur parameter
...
latest graphicsmagick is required
2016-05-21 09:54:12 +09:00
nagadomi
145b47dbf5
Add use_transparent_png option
2016-05-19 23:02:02 +09:00
nagadomi
8d3950b90a
Change the default parameter (epoch, downsampling_filters)
2016-05-15 11:33:34 +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
7af5c9443d
Add model option and 12 layers net
2016-04-23 09:19:03 +09:00
nagadomi
8d6451a51b
Add -max_training_image_size option
2016-04-10 23:06:39 +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
7f448a98c4
Change default validation_crops(160)
2016-03-22 10:19:52 +09:00
nagadomi
cef98be88f
Change default downsampling_filters (Box,Catrom)
2016-03-22 10:18:59 +09:00
nagadomi
8a799e2d56
Add support for noise level 3
2016-03-18 15:47:53 +09:00
nagadomi
d4833160c7
Optionalize downsampling filters
2016-03-17 17:58:37 +09:00
nagadomi
4d115e4bdb
Add support for plotting loss chart
2016-03-14 05:06:14 +09:00
nagadomi
947746b32b
Change default parameter
2015-12-11 16:31:24 +09:00
nagadomi
9f935835dd
Add -save_history option
2015-12-04 18:49:34 +09:00
nagadomi
aaac6ed6e5
Refactor training loop
...
more shuffle
2015-11-30 17:18:52 +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
42bd89151e
Add -gpu option in train.lua
2015-11-13 19:26:58 +09:00
nagadomi
2d5a090305
Change default -nr_rate
2015-11-10 22:07:45 +09:00
nagadomi
3ea16b3b86
tunable parameters
2015-11-07 07:18:22 +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
243d8821be
Add -backend option to use cuDNN
2015-10-31 04:38:28 +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
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
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
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
2231423056
update training script
2015-05-17 14:43:07 +09:00
nagadomi
1273b3609e
first commit
2015-05-16 14:48:05 +09:00