Home Explore Help
Sign In
hugh
/
waifu2x
1
0
Fork 0
Files Issues 0 Pull Requests 0 Wiki
Tree: b89833daa3
Branches Tags
waifu2x
/
tools
/
export.sh

export.sh 916 B
History Raw

123456789101112131415161718192021 
  1. #!/bin/sh -x
    2. 

  2. 

  3. export_model() {
    4. 

  4.     if [ -f models/${1}/scale2.0x_model.t7 ] && [ ! -h models/${1}/scale2.0x_model.t7 ] ; then
    5. 

  5. 	th tools/export_model.lua -i models/${1}/scale2.0x_model.t7 -o models/${1}/scale2.0x_model.json
    6. 

  6.     fi
    7. 

  7.     if [ -f models/${1}/noise1_model.t7 ] && [ ! -h models/${1}/noise1_model.t7 ]; then
    8. 

  8. 	th tools/export_model.lua -i models/${1}/noise1_model.t7 -o models/${1}/noise1_model.json
    9. 

  9.     fi
    10. 

  10.     if [ -f models/${1}/noise2_model.t7 ] && [ ! -h models/${1}/noise2_model.t7 ]; then
    11. 

  11. 	th tools/export_model.lua -i models/${1}/noise2_model.t7 -o models/${1}/noise2_model.json
    12. 

  12.     fi
    13. 

  13.     if [ -f models/${1}/noise3_model.t7 ] && [ ! -h models/${1}/noise3_model.t7 ]; then
    14. 

  14. 	th tools/export_model.lua -i models/${1}/noise3_model.t7 -o models/${1}/noise3_model.json
    15. 

  15.     fi
    16. 

  16. }
    17. 

  17. export_model vgg_7/art
    18. 

  18. export_model vgg_7/art_y
    19. 

  19. export_model vgg_7/photo
    20. 

  20. export_model vgg_7/ukbench
    21. 

  21. export_model upconv_7/art
    22.