nagadomi
61aeb46303
Use bilinear instead of bicubic beacuse of overshoot
2016-10-22 15:11:05 +09:00
nagadomi
b066761cdc
Add data augmentation for user method
2016-10-21 18:34:48 +09:00
nagadomi
a72af8cfef
Use 'image' module because the deadlock problem has fixed
2016-10-21 01:21:42 +09:00
nagadomi
1e80e45a03
add experimental models
2016-10-20 23:41:39 +09:00
nagadomi
50fd999c38
Add -gpu option
2016-10-18 19:54:00 +09:00
nagadomi
a604aa3b7a
Add some experimental model
2016-10-08 17:21:01 +09:00
nagadomi
e71ff1fc6a
Fix a bug in random blur
2016-10-05 18:48:32 +09:00
nagadomi
3e77378983
Add SSIMCriterion (supports only forward())
2016-10-02 22:22:50 +09:00
nagadomi
d7ab10581c
Use clamp
2016-09-24 08:17:37 +09:00
nagadomi
f1a45b19c5
test
2016-09-24 07:46:50 +09:00
nagadomi
0517605d0d
Remove flip() from preprocess() because the data will flip() at flip_augmentation()
2016-09-24 06:22:30 +09:00
nagadomi
f65132dadb
Use conv2d instead of nn.SpatialConvolutionMM
2016-09-24 05:54:18 +09:00
nagadomi
5a3d012f4e
Add random blur method for data augmentation
2016-09-24 05:32:33 +09:00
nagadomi
cabeeed2a7
Add Clip(0,1) to last layer
2016-09-14 21:23:29 +09:00
nagadomi
f11108cba1
Add comment about RNG seed
2016-09-14 21:19:59 +09:00
nagadomi
451ee1407f
Stop calculate the instance loss when oracle_rate=0
2016-09-11 20:59:32 +09:00
nagadomi
c2e4bb4380
Support for multi-thread in training
...
And remove `sys`,`image` and `graphicsmagicks.conveter` from the training code because those causes the deadlock on thread package.
2016-09-11 06:31:44 +09:00
nagadomi
f7e83e4465
performance tuning
2016-09-08 22:35:27 +09:00
nagadomi
382d493514
no confidence change for #128
2016-07-27 22:55:56 +09:00
nagadomi
b3c968ef11
Fix a bug in user method when scale=2
2016-07-20 14:06:17 +09:00
nagadomi
9563d84302
No bias at FullConvolution
2016-07-12 11:15:12 +09:00
nagadomi
81df729a8a
Remove -loss option
2016-07-09 15:05:11 +09:00
nagadomi
a0d8826779
Merge branch 'master' into upconv
2016-07-09 09:58:48 +09:00
nagadomi
8e90a44d30
noise0
2016-07-09 09:08:04 +09:00
nagadomi
06e073089b
Fix for edac608f
2016-07-06 08:04:24 +09:00
nagadomi
14330e919c
tuning a little
2016-07-06 08:03:24 +09:00
nagadomi
deae802267
Fix a bug in noise reduction when using Y model
2016-07-06 00:47:58 +09:00
nagadomi
edac608f18
Add support for user specified pairwise data for universal filter
2016-07-05 02:42:40 +09:00
nagadomi
9ec1f5159b
Fix a bug in ClippedMSECriterion
2016-07-04 11:16:04 +09:00
nagadomi
eeb90dc39b
Fix a bug in preprocessing when filters=box only
2016-06-22 00:37:29 +09:00
nagadomi
d8fcde46f3
Change default -validation_crops to 200.
2016-06-20 15:56:26 +09:00
nagadomi
fb3636c3b7
Change noise rate in jpeg training
2016-06-20 15:55:43 +09:00
nagadomi
f61d59d28a
Change default params. enabling jpeg_chroma_subsampling by default
2016-06-18 20:27:34 +09:00
nagadomi
9d3e1a241e
refactor
2016-06-18 20:25:15 +09:00
nagadomi
2e26122a9e
performance tuning
2016-06-17 20:40:03 +09:00
nagadomi
e50c7d1478
Remove unused modules
2016-06-13 12:48:11 +09:00
nagadomi
3b1b2d03fe
Change default -nr_rate to 0.65
2016-06-13 12:44:32 +09:00
nagadomi
25e293202a
Add support for tta_level=1; Add support for TTA to web.lua
2016-06-12 16:55:05 +09:00
nagadomi
af74a67bd1
Add -force_cudnn option; support for cuDNN in waifu2x.lua/web.lua
2016-06-12 16:33:50 +09:00
nagadomi
6be1479710
Fix a block_size issue when using upconv_7 model
2016-06-12 05:15:24 +09:00
nagadomi
9103d393fe
Fix a performance issue
2016-06-12 05:13:40 +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
afac4b52ab
Add -batch_size option to waifu2x.lua/web.lua
2016-06-09 14:03:18 +09:00
nagadomi
0b949c05a7
Add support for TTA level
2016-06-09 13:09:28 +09:00
nagadomi
9514027f65
Fix a bug that -nr_rate is not used
2016-06-09 02:44:22 +09:00
nagadomi
e5cfd3dfce
Add -resume option
2016-06-09 02:39:52 +09:00
nagadomi
51914b894a
change weight initialization and upconv_7
2016-06-08 06:58:46 +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
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
958dcfaeb8
Add more informative message for CUDA error
2016-05-06 21:48:53 +09:00
nagadomi
5bfe504de1
Add error message when the CUDA module loading fails
2016-05-06 02:20:54 +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