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
nagadomi
2305e31616
Merge branch 'master' into photo
2015-12-02 06:54:37 +09:00
nagadomi
03a17a636b
Merge branch 'master' of github.com:nagadomi/waifu2x
2015-12-02 06:54:22 +09:00
nagadomi
86feb1d4c9
Change the sampling method
2015-12-02 06:53:26 +09:00
nagadomi
9727bba51a
Fix gamma correction
2015-12-01 20:56:56 +09:00
nagadomi
5cda0b71d7
Fix weird value in alpha channel
2015-12-01 20:42:44 +09:00
nagadomi
c5df72439f
Fix cuda tensor error in async environment
2015-12-01 18:21:06 +09:00
nagadomi
d2c081bbcf
Improve alpha channel handling #29
...
- make border
- scale the alpha channel by waifu2x
- composite
2015-12-01 17:26:45 +09:00
nagadomi
a975155828
Change downscaling filters when photo training
2015-11-30 17:19:50 +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
5a80a70e65
Fix error when image has alpha channel
2015-11-21 17:24:24 +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
f7b298690d
Fix the missing initialization function
...
I don't know why was this function removed.
2015-11-18 21:46:43 +09:00
nagadomi
b5db84d42e
Change the jpeg config for the photo model
2015-11-15 09:36:40 +09:00
nagadomi
42bd89151e
Add -gpu option in train.lua
2015-11-13 19:26:58 +09:00
nagadomi
c345f35ed0
Use Image.toString instead of Image.toBlob
2015-11-13 16:11:12 +09:00
nagadomi
3c1c11d88e
directly load float data
2015-11-13 15:53:41 +09:00
nagadomi
28cc864c0c
Fix NaN bug in iproc.white_noise
2015-11-12 08:03:24 +09:00
nagadomi
b63e08905b
Remove unused variable
2015-11-11 10:54:28 +09:00
nagadomi
4a4885c856
Add -white_noise option
...
post-processing for anime screencap.
2015-11-11 10:42:33 +09:00
nagadomi
0941d098e1
more round
2015-11-10 22:38:22 +09:00
nagadomi
2d5a090305
Change default -nr_rate
2015-11-10 22:07:45 +09:00
nagadomi
eaeaef10b7
more flip
2015-11-09 23:45:01 +09:00
nagadomi
af1b9c604b
refactor
2015-11-09 04:35:11 +09:00