nagadomi
490eb33a6b
Minimize the weighted huber loss instead of the weighted mean square error
...
Huber loss is less sensitive to outliers(i.e. noise) in data than the squared error loss.
2015-10-31 22:05:59 +09:00
nagadomi
243d8821be
Add -backend option to use cuDNN
2015-10-31 04:38:28 +09:00
nagadomi
9ddee6088d
Add WeightedHuberCriterion
2015-10-31 04:36:20 +09:00
nagadomi
9d63ed2947
Fix gradient scale in WeightedMSECriterion
2015-10-31 04:35:33 +09:00
nagadomi
4c691b4640
refactor
2015-10-29 22:44:15 +09:00
nagadomi
21ea5dd858
add cudnn converter
2015-10-29 18:05:33 +09:00
nagadomi
da786e15ba
remove noise_scale training
2015-10-28 16:27:31 +09:00
nagadomi
3abc5a03e3
refactor
2015-10-28 16:01:07 +09:00
nagadomi
956ae0d24c
merge from master
2015-10-26 09:39:41 +09:00
nagadomi
8dea362bed
sync from internal repo
...
- 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.
2015-10-26 09:23:52 +09:00
Kenta Murata
306ee3c76d
Fix typo
2015-08-30 10:47:20 +09:00
nagadomi
ee826715b7
explicitly loading torch/cutorch
2015-08-05 16:20:45 +09:00
nagadomi
2ffa93213e
remove exception handling for cunn
2015-08-05 16:03:51 +09:00
nagadomi
dcd43b89fd
fix missing nn in latest torch7
2015-08-05 11:49:38 +09:00
nagadomi
54580ba8c0
add data augmentation method that uses overlay
2015-08-02 22:02:14 +09:00
nagadomi
6c65463564
fix slow compression
2015-07-25 02:39:36 +09:00
nagadomi
e5a2ef97c7
set compression level of the output file
2015-07-24 17:16:27 +09:00
nagadomi
8e1625d001
change the noisy:cleaned ratio of training data
2015-07-18 11:17:34 +09:00
nagadomi
e3d3a8355c
change training script
...
- add AlexNet's color noise (default: false)
- add `photo` category for noise level setting
2015-07-11 21:57:04 +09:00
nagadomi
d8ca857a82
change initial weights
2015-07-11 14:52:51 +09:00
nagadomi
fd9dadd7a4
use settings.validation_crop
2015-06-26 20:12:51 +09:00
nagadomi
5b4d692f03
add support for RGB color space reconstruction
...
- 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 ...
2015-06-23 02:55:30 +09:00
nagadomi
628bd971c9
Merge branch 'master' of github.com:nagadomi/waifu2x into rgb
2015-06-22 23:04:17 +09:00
nagadomi
04c1d7f936
remove unused function
2015-06-22 23:03:14 +09:00
nagadomi
1cdceb1648
throw error when failed to load image
2015-06-22 23:01:56 +09:00
nagadomi
bbc3cac273
add support for transparent png (retain the alpha channel)
2015-06-16 20:41:48 +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
nagadomi
1c1f2292d1
fix ugly chroma scaling
2015-05-22 20:06:25 +09:00
nagadomi
0d115f46f6
fix data augmentation in pairwise_transform.scale
2015-05-22 18:59:25 +09:00
nagadomi
ffd52fdb2b
super typo fix
2015-05-19 16:47:52 +09:00
nagadomi
2231423056
update training script
2015-05-17 14:43:07 +09:00
nagadomi
1273b3609e
first commit
2015-05-16 14:48:05 +09:00
