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

3 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 7856ba90c7 Update trained models 2015-11-06 10:43:05 +09:00
nagadomi 95e9d276a3 add new model for photo
- add ukbench model (models/ukbench)
- ukbench model was trained by Recognition Benchmark Images (http://www.vis.uky.edu/~stewe/ukbench/)

usage:
$ th waifu2x.lua -m scale -model_dir models/ukbench -i input.jpg -o output.png
2015-06-23 18:03:23 +09:00