1
0
Fork 0
mirror of synced 2024-09-21 20:11:38 +12:00
Commit graph

14 commits

Author SHA1 Message Date
nagadomi
f0fc2c89d1 Add support for grayscale data 2017-04-09 20:53:53 +09:00
nagadomi
bdff13a35f Use rgb2y instead of rgb2yuv 2016-11-06 05:32:36 +09:00
nagadomi
a72af8cfef Use 'image' module because the deadlock problem has fixed 2016-10-21 01:21:42 +09:00
nagadomi
c2e4bb4380 Support for multi-thread in training
And remove `sys`,`image` and `graphicsmagicks.conveter` from the training code because those causes the deadlock on thread package.
2016-09-11 06:31:44 +09:00
nagadomi
f7e83e4465 performance tuning 2016-09-08 22:35:27 +09:00
nagadomi
9d3e1a241e refactor 2016-06-18 20:25:15 +09:00
nagadomi
b8ff8c6787 Remove -gamma_correction option 2016-06-10 07:37:39 +09:00
nagadomi
01b2e6d441 Remove -upsampling_filter option 2016-06-10 07:34:11 +09:00
nagadomi
d0630d3a20 individual filters and box-only support 2016-06-06 14:04:13 +09:00
nagadomi
70eb2b508f Fix a performance problem in resampling 2016-05-30 19:15:54 +09:00
nagadomi
99e6dd1a57 Fix border removing 2016-05-28 10:25:08 +09:00
nagadomi
7814691cbf Add resize_blur parameter
latest graphicsmagick is required
2016-05-21 09:54: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
51ae485cd1 Add new models
upconv_7 is 2.3x faster than previous model
2016-05-13 09:49:53 +09:00