- 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.
- 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.
- 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 ...
- 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
