nagadomi
af1b9c604b
refactor
2015-11-09 04:35:11 +09:00
nagadomi
b335f3a9ad
Add -tta option
...
The TTA mode:
- 8x slower than normal mode
- improves PSNR +0.1
2015-11-09 04:01:28 +09:00
nagadomi
7d67afb982
Use iproc.crop instead of image.crop
2015-11-08 22:11:36 +09:00
nagadomi
6e41796328
Use gamma 2.2 space when resizing
2015-11-08 20:28:14 +09:00
nagadomi
180e7676e4
Drop full transparent background
2015-11-08 06:32:15 +09:00
nagadomi
797b45ae23
Use roundf-like clip for 8 bit-depth image
...
Maybe PSNR +0.03 improved by this commit
2015-11-08 05:44:14 +09:00
nagadomi
bd63f99b59
clip
2015-11-08 01:21:52 +09:00
nagadomi
3b2aa67e55
Improve web interface
2015-11-07 22:54:29 +09:00
nagadomi
15f3f2d901
typo
2015-11-07 07:38:45 +09:00
nagadomi
84c38baab7
Fix double loading of image
2015-11-07 07:33:41 +09:00
nagadomi
3ea16b3b86
tunable parameters
2015-11-07 07:18:22 +09:00
nagadomi
cf862782a5
depth 8
2015-11-07 06:39:47 +09:00
nagadomi
c773e18e59
Add trade-off parameter for noise reduction
2015-11-07 06:37:53 +09:00
nagadomi
43832a9f92
Add support for depth of PNG
2015-11-06 13:12:36 +09:00
nagadomi
539941c234
Add support for url cache in web.lua
2015-11-06 12:24:27 +09:00
nagadomi
903d945652
cleanup
2015-11-06 10:08:54 +09:00
nagadomi
44208f18ab
Fix handling for gamma embed PNG
2015-11-05 14:59:51 +09:00
nagadomi
bbd0ff39ad
Fix handling for CMYK JPEG
2015-11-03 23:20:21 +09:00
nagadomi
b35a9ae7d7
tuning
2015-11-03 06:10:44 +09:00
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