1
0
Fork 0
mirror of synced 2024-06-29 11:41:29 +12:00
Commit graph

109 commits

Author SHA1 Message Date
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
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