1
0
Fork 0
mirror of synced 2024-05-19 20:32:22 +12:00
Commit graph

6 commits

Author SHA1 Message Date
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 f221febf5f Update jsons 2016-03-27 17:47:19 +09:00
nagadomi 69ded0223a Add level 3 noise reduction models 2016-03-27 17:26:10 +09:00
nagadomi 4742dd1807 Update anime_style_art(Y) models 2016-03-27 17:15:14 +09:00
nagadomi a897a83f31 Fix loading error in Y model 2016-01-23 09:36:54 +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