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

10 commits

Author SHA1 Message Date
nagadomi 929c7f85a9 Support Linear module 2018-06-05 03:57:31 +09:00
nagadomi 0b1a13d9c0 Add to json formt model; Fix json format; 2018-04-28 11:00:13 +09:00
nagadomi cf2493a1a4 Add support for resnet_14l in tools/export.lua; Add resnet_14l.prototxt 2017-01-26 05:43:55 +09:00
nagadomi 9563d84302 No bias at FullConvolution 2016-07-12 11:15:12 +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 29dc96eadf Remove unused attributes in json 2016-03-27 17:37:38 +09:00
nagadomi 986ef34ffc Add color and gamma attribute to json file 2015-11-08 22:12:48 +09:00
nagadomi 5ab2c605c8 Fix header include order issues 2015-11-08 18:31:46 +09:00
nagadomi 268b9f6671 refactor tools/export_model.lua 2015-11-06 10:42:37 +09:00
nagadomi 3abc5a03e3 refactor 2015-10-28 16:01:07 +09:00
Renamed from export_model.lua (Browse further)