 nagadomi
|
7af5c9443d
Add model option and 12 layers net
|
9 vuotta sitten |
|
nagadomi
|
68a6d4cef5
Use MSE instead of PSNR
|
9 vuotta sitten |
|
nagadomi
|
fa9355be7c
Fix validation metric
|
9 vuotta sitten |
|
nagadomi
|
a92b2cb386
Fix progressbar
|
9 vuotta sitten |
|
nagadomi
|
ea780f1871
Fix validation metric
|
9 vuotta sitten |
|
nagadomi
|
30fe5db735
Add upsampling_filter option
|
9 vuotta sitten |
|
nagadomi
|
13f702b968
Add support for resizing with gamma correction
|
9 vuotta sitten |
|
nagadomi
|
d4833160c7
Optionalize downsampling filters
|
9 vuotta sitten |
|
nagadomi
|
4d115e4bdb
Add support for plotting loss chart
|
9 vuotta sitten |
|
nagadomi
|
459c7c5e18
Use similar distribution on train/validation
|
9 vuotta sitten |
|
nagadomi
|
1900ac7500
Use PSNR for evaluation
|
9 vuotta sitten |
|
nagadomi
|
664322fa97
Remove cleanup_model.lua; Change model format from binary to ascii
|
9 vuotta sitten |
|
nagadomi
|
290a5f960b
Use Huber loss instead of MSE
|
9 vuotta sitten |
|
nagadomi
|
9b238bd693
Use clearState()
|
9 vuotta sitten |
|
nagadomi
|
9f935835dd
Add -save_history option
|
10 vuotta sitten |
|
nagadomi
|
aaac6ed6e5
Refactor training loop
|
10 vuotta sitten |
|
nagadomi
|
c72ec3112b
Add -random_unsharp_mask_rate option for photo
|
10 vuotta sitten |
|
nagadomi
|
d8ba661d6b
Add -jpeg_chroma_subsampling_rate option for JPEG denoise training
|
10 vuotta sitten |
|
nagadomi
|
42bd89151e
Add -gpu option in train.lua
|
10 vuotta sitten |
|
nagadomi
|
5ab2c605c8
Fix header include order issues
|
10 vuotta sitten |
|
nagadomi
|
797b45ae23
Use roundf-like clip for 8 bit-depth image
|
10 vuotta sitten |
|
nagadomi
|
3ea16b3b86
tunable parameters
|
10 vuotta sitten |
|
nagadomi
|
c773e18e59
Add trade-off parameter for noise reduction
|
10 vuotta sitten |
|
nagadomi
|
903d945652
cleanup
|
10 vuotta sitten |
|
nagadomi
|
b35a9ae7d7
tuning
|
10 vuotta sitten |
|
nagadomi
|
490eb33a6b
Minimize the weighted huber loss instead of the weighted mean square error
|
10 vuotta sitten |
|
nagadomi
|
da786e15ba
remove noise_scale training
|
10 vuotta sitten |
|
nagadomi
|
3abc5a03e3
refactor
|
10 vuotta sitten |
|
nagadomi
|
8dea362bed
sync from internal repo
|
10 vuotta sitten |
|
nagadomi
|
54580ba8c0
add data augmentation method that uses overlay
|
10 vuotta sitten |
