export_all.sh 2.0 KB

123456789101112131415161718192021222324252627282930313233343536373839
  1. #!/bin/sh -x
  2. export_model() {
  3. if [ -f models/${1}/scale2.0x_model.t7 ] && [ ! -h models/${1}/scale2.0x_model.t7 ] ; then
  4. th tools/export_model.lua -i models/${1}/scale2.0x_model.t7 -o models/${1}/scale2.0x_model.json
  5. fi
  6. if [ -f models/${1}/noise0_model.t7 ] && [ ! -h models/${1}/noise0_model.t7 ]; then
  7. th tools/export_model.lua -i models/${1}/noise0_model.t7 -o models/${1}/noise0_model.json
  8. fi
  9. if [ -f models/${1}/noise1_model.t7 ] && [ ! -h models/${1}/noise1_model.t7 ]; then
  10. th tools/export_model.lua -i models/${1}/noise1_model.t7 -o models/${1}/noise1_model.json
  11. fi
  12. if [ -f models/${1}/noise2_model.t7 ] && [ ! -h models/${1}/noise2_model.t7 ]; then
  13. th tools/export_model.lua -i models/${1}/noise2_model.t7 -o models/${1}/noise2_model.json
  14. fi
  15. if [ -f models/${1}/noise3_model.t7 ] && [ ! -h models/${1}/noise3_model.t7 ]; then
  16. th tools/export_model.lua -i models/${1}/noise3_model.t7 -o models/${1}/noise3_model.json
  17. fi
  18. if [ -f models/${1}/noise0_scale2.0x_model.t7 ] && [ ! -h models/${1}/noise0_scale2.0x_model.t7 ]; then
  19. th tools/export_model.lua -i models/${1}/noise0_scale2.0x_model.t7 -o models/${1}/noise0_scale2.0x_model.json
  20. fi
  21. if [ -f models/${1}/noise1_scale2.0x_model.t7 ] && [ ! -h models/${1}/noise1_scale2.0x_model.t7 ]; then
  22. th tools/export_model.lua -i models/${1}/noise1_scale2.0x_model.t7 -o models/${1}/noise1_scale2.0x_model.json
  23. fi
  24. if [ -f models/${1}/noise2_scale2.0x_model.t7 ] && [ ! -h models/${1}/noise2_scale2.0x_model.t7 ]; then
  25. th tools/export_model.lua -i models/${1}/noise2_scale2.0x_model.t7 -o models/${1}/noise2_scale2.0x_model.json
  26. fi
  27. if [ -f models/${1}/noise3_scale2.0x_model.t7 ] && [ ! -h models/${1}/noise3_scale2.0x_model.t7 ]; then
  28. th tools/export_model.lua -i models/${1}/noise3_scale2.0x_model.t7 -o models/${1}/noise3_scale2.0x_model.json
  29. fi
  30. }
  31. export_model vgg_7/art
  32. export_model upconv_7/art
  33. export_model upconv_7l/art
  34. export_model vgg_7/photo
  35. export_model upconv_7/photo
  36. export_model upconv_7l/photo