nagadomi
c5df72439f
Fix cuda tensor error in async environment
2015-12-01 18:21:06 +09:00
nagadomi
d2c081bbcf
Improve alpha channel handling #29
...
- make border
- scale the alpha channel by waifu2x
- composite
2015-12-01 17:26:45 +09:00
nagadomi
a975155828
Change downscaling filters when photo training
2015-11-30 17:19:50 +09:00
nagadomi
aaac6ed6e5
Refactor training loop
...
more shuffle
2015-11-30 17:18:52 +09:00
nagadomi
c72ec3112b
Add -random_unsharp_mask_rate option for photo
2015-11-27 18:36:36 +09:00
nagadomi
d8ba661d6b
Add -jpeg_chroma_subsampling_rate option for JPEG denoise training
2015-11-26 17:10:57 +09:00
nagadomi
5a80a70e65
Fix error when image has alpha channel
2015-11-21 17:24:24 +09:00
nagadomi
abedec9d9b
Fix filter name for graphicsmagick
...
graphicsmagick package has bug in the definition of FilteTypes.
I will send PR to fix that bug.
This commit is related to it.
You should upgrade graphicsmagick package.
$ luarocks install graphicsmagick
2015-11-21 11:57:33 +09:00
nagadomi
f7b298690d
Fix the missing initialization function
...
I don't know why was this function removed.
2015-11-18 21:46:43 +09:00
nagadomi
b5db84d42e
Change the jpeg config for the photo model
2015-11-15 09:36:40 +09:00
nagadomi
42bd89151e
Add -gpu option in train.lua
2015-11-13 19:26:58 +09:00
nagadomi
c345f35ed0
Use Image.toString instead of Image.toBlob
2015-11-13 16:11:12 +09:00
nagadomi
3c1c11d88e
directly load float data
2015-11-13 15:53:41 +09:00
nagadomi
28cc864c0c
Fix NaN bug in iproc.white_noise
2015-11-12 08:03:24 +09:00
nagadomi
b63e08905b
Remove unused variable
2015-11-11 10:54:28 +09:00
nagadomi
4a4885c856
Add -white_noise option
...
post-processing for anime screencap.
2015-11-11 10:42:33 +09:00
nagadomi
0941d098e1
more round
2015-11-10 22:38:22 +09:00
nagadomi
2d5a090305
Change default -nr_rate
2015-11-10 22:07:45 +09:00
nagadomi
eaeaef10b7
more flip
2015-11-09 23:45:01 +09:00
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
nagadomi
1273b3609e
first commit
2015-05-16 14:48:05 +09:00
