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
0349fc774c
refactor
2016-06-02 10:12:04 +09:00
nagadomi
c89fd7249a
Add learning_rate_decay
2016-06-02 10:11:15 +09:00
nagadomi
70eb2b508f
Fix a performance problem in resampling
2016-05-30 19:15:54 +09:00
nagadomi
70a2849e39
Fix missing file
2016-05-30 06:48:26 +09:00
nagadomi
99e6dd1a57
Fix border removing
2016-05-28 10:25:08 +09:00
nagadomi
8a65db7bab
Change the evaluation metric
2016-05-27 16:57:14 +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
f6a37b66c3
Add support for Transparent PNG in convert_data.lua
2016-05-15 12:34:03 +09:00
nagadomi
8d3950b90a
Change the default parameter (epoch, downsampling_filters)
2016-05-15 11:33:34 +09:00
nagadomi
c028ce6e4f
Fix a bug in reconstrct.scale() when inner_scale > 1 && y only model
2016-05-15 11:31:14 +09:00
nagadomi
a210090033
Convert model files; Add new pretrained model
...
- Add new pretrained model to ./models/upconv_7
- Move old models to ./models/vgg_7
- Use nn.LeakyReLU instead of w2nn.LeakyReLU
- Add useful attribute to .json
New JSON attribute:
The first layer has `model_config` attribute.
It contains:
model_arch: architecture name of model. see `lib/srcnn.lua`
scale_factor: if scale_factor > 1, model:forward() changes image resolution with scale_factor.
channels: input/output channels. if channels == 3, model is RGB model.
offset: pixel size that is to be removed from output.
for example:
(scale_factor=1, offset=7, input=100x100) => output=(100-7)x(100-7)
(scale_factor=2, offset=12, input=100x100) => output=(100*2-12)x(100*2-12)
And each layer has `class_name` attribute.
2016-05-15 03:04:08 +09:00
nagadomi
48411a4dde
refactor
2016-05-14 16:51: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
e62305377f
Add compression.size()
2016-05-13 09:35:53 +09:00
nagadomi
8da52d5fb9
Merge from master
2016-04-23 12:48:24 +09:00
nagadomi
7af5c9443d
Add model option and 12 layers net
2016-04-23 09:19:03 +09:00
nagadomi
da03209d3e
Remove PNG compression option
2016-04-20 18:53:31 +09:00
nagadomi
68a6d4cef5
Use MSE instead of PSNR
...
PSNR depends on the minibatch size and those group.
2016-04-17 02:08:38 +09:00
nagadomi
3a27e122ac
Add support for grayscale output
2016-04-15 13:29:50 +09:00
nagadomi
ba2fe8d9c1
Improve alpha channel detection
...
latest `graphicsmagick` is required.
2016-04-15 13:14:10 +09:00
nagadomi
fbad30c031
Fix embed gamma handling
2016-04-15 09:13:37 +09:00
nagadomi
1407973b88
Use the specified filter
2016-04-12 01:33:17 +09:00
nagadomi
a938cd5994
Reduce draw calls
2016-04-10 23:30:23 +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
fd06a47d56
Fix clearState
2016-03-28 18:38:31 +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
86ad50f7cd
Merge branch 'master' of github.com:nagadomi/waifu2x into dev
2016-03-21 03:59:17 +09:00
nagadomi
57e0f52b41
Reduce memory usage
2016-03-21 03:42:47 +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
4d115e4bdb
Add support for plotting loss chart
2016-03-14 05:06:14 +09:00
nagadomi
5590674783
Fix NaN in PSNRCriterion
2016-03-12 08:47:14 +09:00
nagadomi
4a1629d046
More clearState for nn.SpatialConvolutionMM
2016-03-12 07:25:16 +09:00
nagadomi
223dcead67
Add support for clearState() in w2nn.LeakyReLU
2016-03-12 06:59:23 +09:00
nagadomi
1900ac7500
Use PSNR for evaluation
2016-03-12 06:53:42 +09:00
nagadomi
41581a0d55
Remove cleanup_model
2016-03-12 05:43:04 +09:00
nagadomi
9b238bd693
Use clearState()
2016-03-11 11:12:02 +09:00
nagadomi
7708c7e75a
Merge branch 'master' of github.com:nagadomi/waifu2x
2015-12-11 16:35:00 +09:00
nagadomi
947746b32b
Change default parameter
2015-12-11 16:31:24 +09:00
nagadomi
56582b8401
Fix error in training script due to missing libcudnn.so
2015-12-09 07:04:04 +09:00
nagadomi
9f935835dd
Add -save_history option
2015-12-04 18:49:34 +09:00
nagadomi
eea4c31d7b
Change parameters
2015-12-04 18:47:33 +09:00