1
0
Fork 0
mirror of synced 2024-06-26 18:20:26 +12:00
Commit graph

13 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 7af5c9443d Add model option and 12 layers net 2016-04-23 09:19:03 +09:00
nagadomi fd06a47d56 Fix clearState 2016-03-28 18:38:31 +09:00
nagadomi 4a1629d046 More clearState for nn.SpatialConvolutionMM 2016-03-12 07:25:16 +09:00
nagadomi 56582b8401 Fix error in training script due to missing libcudnn.so 2015-12-09 07:04:04 +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 21ea5dd858 add cudnn converter 2015-10-29 18:05:33 +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 d8ca857a82 change initial weights 2015-07-11 14:52: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 1273b3609e first commit 2015-05-16 14:48:05 +09:00