export.sh 916 B

123456789101112131415161718192021
  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}/noise1_model.t7 ] && [ ! -h models/${1}/noise1_model.t7 ]; then
  7. th tools/export_model.lua -i models/${1}/noise1_model.t7 -o models/${1}/noise1_model.json
  8. fi
  9. if [ -f models/${1}/noise2_model.t7 ] && [ ! -h models/${1}/noise2_model.t7 ]; then
  10. th tools/export_model.lua -i models/${1}/noise2_model.t7 -o models/${1}/noise2_model.json
  11. fi
  12. if [ -f models/${1}/noise3_model.t7 ] && [ ! -h models/${1}/noise3_model.t7 ]; then
  13. th tools/export_model.lua -i models/${1}/noise3_model.t7 -o models/${1}/noise3_model.json
  14. fi
  15. }
  16. export_model vgg_7/art
  17. export_model vgg_7/art_y
  18. export_model vgg_7/photo
  19. export_model vgg_7/ukbench
  20. export_model upconv_7/art