Home Explore Help
Sign In
hugh
/
waifu2x
1
0
Fork 0
Files Issues 0 Pull Requests 0 Wiki
Tree: 451ee1407f
Branches Tags
waifu2x
/
appendix
/
train_upconv_7_art.sh

train_upconv_7_art.sh 2.3 KB
History Raw

12345678910111213141516171819202122 
  1. #!/bin/sh
    2. 

  2. 

  3. # convert data 
    4. 

  4. th convert_data.lua -max_training_image_size 1600
    5. 

  5. 

  6. # scale
    7. 

  7. # after train, I selected `scale2.0x_model.86-1.t7`, because seemingly this model is best mesh-artifact free model. Maybe visual testing is required.
    8. 

  8. # ref: https://github.com/nagadomi/waifu2x/issues/125
    9. 

  9. th train.lua -save_history 1 -scale 2 -model upconv_7 -method scale -model_dir models/test/upconv_7_rev5 -downsampling_filters "Box,Sinc" -test query/pixel-art-small.png -backend cudnn -thread 4 -oracle_rate 0.0 -inner_epoch 2 -epoch 100
    10. 

  10. 

  11. # noise_scale
    12. 

  12. 

  13. th train.lua -save_history 1 -model upconv_7 -method noise_scale -noise_level 0 -model_dir models/test/upconv_7_rev6 -downsampling_filters "Box,Sinc" -test query/noise_test.jpg -backend cudnn -thread 4  -resume models/test/upconv_7_rev5/scale2.0x_model.t7 -style art
    14. 

  14. th train.lua -save_history 1 -model upconv_7 -method noise_scale -noise_level 1 -model_dir models/test/upconv_7_rev6 -downsampling_filters "Box,Sinc" -test query/noise_test.jpg -backend cudnn -thread 4  -resume models/test/upconv_7_rev5/scale2.0x_model.t7 -style art
    15. 

  15. th train.lua -save_history 1 -model upconv_7 -method noise_scale -noise_level 2 -model_dir models/test/upconv_7_rev6 -downsampling_filters "Box,Sinc" -test query/noise_test.jpg -backend cudnn -thread 4  -resume models/test/upconv_7_rev5/scale2.0x_model.t7 -style art
    16. 

  16. th train.lua -save_history 1 -model upconv_7 -method noise_scale -noise_level 3 -model_dir models/test/upconv_7_rev6 -downsampling_filters "Box,Sinc" -test query/noise_test.jpg -backend cudnn -thread 4  -resume models/test/upconv_7_rev5/scale2.0x_model.t7 -style art -nr_rate 1
    17. 

  17. 

  18. # noise
    19. 

  19. th train.lua -save_history 1 -model vgg_7 -method noise -noise_level 0 -model_dir models/test/yuv420_rev2 -test query/noise_test.jpg -backend cudnn -thread 4 -style art -crop_size 32
    20. 

  20. th train.lua -save_history 1 -model vgg_7 -method noise -noise_level 1 -model_dir models/test/yuv420_rev2 -test query/noise_test.jpg -backend cudnn -thread 3 -style art -crop_size 32
    21. 

  21. th train.lua -save_history 1 -model vgg_7 -method noise -noise_level 2 -model_dir models/test/yuv420_rev2 -test query/noise_test.jpg -backend cudnn -thread 3 -style art -crop_size 32
    22. 

  22. th train.lua -save_history 1 -model vgg_7 -method noise -noise_level 3 -model_dir models/test/yuv420_rev2 -test query/noise_test.jpg -backend cudnn -thread 3 -style art -crop_size 32 -nr_rate 1 
    23.