nagadomi
|
cdafbf00ae
|
Add BCE(binary cross entropy) loss for segmentation
Sigmoid() output is required.
|
2017-02-23 08:48:00 +09:00 |
|
|
nagadomi
|
385020e0e1
|
Fix a bug of 763f5ddcab
|
2017-02-19 01:42:40 +09:00 |
|
|
nagadomi
|
8b5ccbed08
|
Merge branch 'dev' of into dev
|
2017-02-12 17:48:44 +09:00 |
|
|
nagadomi
|
c6e3a68974
|
Optimize flip in user method when -patchs is small
|
2017-02-12 17:48:21 +09:00 |
|
|
nagadomi
|
763f5ddcab
|
Fix crop bug in rare case
|
2017-02-12 17:46:07 +09:00 |
|
|
nagadomi
|
1db2eeb788
|
Merge branch 'master' of github.com:nagadomi/waifu2x into dev
|
2017-02-12 17:27:33 +09:00 |
|
|
nagadomi
|
d6f5e48dab
|
Add support for icc profile
|
2017-02-12 17:19:03 +09:00 |
|
|
nagadomi
|
29260ede24
|
fix batchwise psnr
|
2017-02-12 02:04:23 +09:00 |
|
|
nagadomi
|
b111901cbb
|
add -update_criterion option for back compatible
|
2017-02-12 01:56:03 +09:00 |
|
|
nagadomi
|
3447c6c44c
|
rename srresnet_12l to resnet_14l because it's not the same as SRResNet
|
2017-01-21 16:53:10 +09:00 |
|
|
nagadomi
|
0eccbc6555
|
Add support for MSE loss
|
2017-01-09 13:00:04 +09:00 |
|
|
nagadomi
|
6051dad793
|
Fix a bug in ClippedMSECriterion:backward
|
2017-01-09 12:55:50 +09:00 |
|
|
nagadomi
|
742c9fdd95
|
disable gcn
|
2017-01-09 12:51:45 +09:00 |
|
|
nagadomi
|
43a9b58fcb
|
Add GCN option for user method
|
2016-12-25 20:17:47 +09:00 |
|
|
nagadomi
|
d2cfb8f104
|
Add L1 criterion. Change the criterion of updating model
|
2016-12-05 10:32:26 +09:00 |
|
|
nagadomi
|
61f2abd632
|
remove unused code
|
2016-11-09 03:10:54 +09:00 |
|
|
nagadomi
|
f088af9606
|
add srresnet_12l
|
2016-11-09 03:06:57 +09:00 |
|
|
nagadomi
|
c8271af3ab
|
add srresnet_12l
|
2016-11-09 03:06:26 +09:00 |
|
|
nagadomi
|
9a13392f91
|
Change the default -random_blur_sigma_max 0.75->1.0
|
2016-11-06 18:14:27 +09:00 |
|
|
nagadomi
|
bdff13a35f
|
Use rgb2y instead of rgb2yuv
|
2016-11-06 05:32:36 +09:00 |
|
|
nagadomi
|
4be1141292
|
Fix srresnet_2x
|
2016-11-06 01:19:20 +09:00 |
|
|
nagadomi
|
bdaca16c67
|
perfomance tuning
|
2016-11-02 23:42:14 +09:00 |
|
|
nagadomi
|
6220afc31a
|
Fix a bug in random negate
|
2016-10-24 17:23:09 +09:00 |
|
|
nagadomi
|
74ed227f48
|
Update fcn_v1
|
2016-10-24 09:10:17 +09:00 |
|
|
nagadomi
|
f16950438c
|
Fix tta-mode
|
2016-10-24 09:06:27 +09:00 |
|
|
nagadomi
|
61aeb46303
|
Use bilinear instead of bicubic beacuse of overshoot
|
2016-10-22 15:11:05 +09:00 |
|
|
nagadomi
|
b066761cdc
|
Add data augmentation for user method
|
2016-10-21 18:34:48 +09:00 |
|
|
nagadomi
|
a72af8cfef
|
Use 'image' module because the deadlock problem has fixed
|
2016-10-21 01:21:42 +09:00 |
|
|
nagadomi
|
1e80e45a03
|
add experimental models
|
2016-10-20 23:41:39 +09:00 |
|
|
nagadomi
|
50fd999c38
|
Add -gpu option
|
2016-10-18 19:54:00 +09:00 |
|
|
nagadomi
|
a604aa3b7a
|
Add some experimental model
|
2016-10-08 17:21:01 +09:00 |
|
|
nagadomi
|
e71ff1fc6a
|
Fix a bug in random blur
|
2016-10-05 18:48:32 +09:00 |
|
|
nagadomi
|
3e77378983
|
Add SSIMCriterion (supports only forward())
|
2016-10-02 22:22:50 +09:00 |
|
|
nagadomi
|
d7ab10581c
|
Use clamp
|
2016-09-24 08:17:37 +09:00 |
|
|
nagadomi
|
f1a45b19c5
|
test
|
2016-09-24 07:46:50 +09:00 |
|
|
nagadomi
|
0517605d0d
|
Remove flip() from preprocess() because the data will flip() at flip_augmentation()
|
2016-09-24 06:22:30 +09:00 |
|
|
nagadomi
|
f65132dadb
|
Use conv2d instead of nn.SpatialConvolutionMM
|
2016-09-24 05:54:18 +09:00 |
|
|
nagadomi
|
5a3d012f4e
|
Add random blur method for data augmentation
|
2016-09-24 05:32:33 +09:00 |
|
|
nagadomi
|
cabeeed2a7
|
Add Clip(0,1) to last layer
|
2016-09-14 21:23:29 +09:00 |
|
|
nagadomi
|
f11108cba1
|
Add comment about RNG seed
|
2016-09-14 21:19:59 +09:00 |
|
|
nagadomi
|
451ee1407f
|
Stop calculate the instance loss when oracle_rate=0
|
2016-09-11 20:59:32 +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
|
382d493514
|
no confidence change for #128
|
2016-07-27 22:55:56 +09:00 |
|
|
nagadomi
|
b3c968ef11
|
Fix a bug in user method when scale=2
|
2016-07-20 14:06:17 +09:00 |
|
|
nagadomi
|
9563d84302
|
No bias at FullConvolution
|
2016-07-12 11:15:12 +09:00 |
|
|
nagadomi
|
81df729a8a
|
Remove -loss option
|
2016-07-09 15:05:11 +09:00 |
|
|
nagadomi
|
a0d8826779
|
Merge branch 'master' into upconv
|
2016-07-09 09:58:48 +09:00 |
|
|
nagadomi
|
8e90a44d30
|
noise0
|
2016-07-09 09:08:04 +09:00 |
|
|
nagadomi
|
06e073089b
|
Fix for edac608f
|
2016-07-06 08:04:24 +09:00 |
|