rebuild.sh 1.0 KB

1234567891011121314151617181920212223
  1. #!/bin/sh -x
  2. # maybe you should backup models
  3. rebuild() {
  4. if [ -f models/${1}/scale2.0x_model.t7 ] && [ ! -h models/${1}/scale2.0x_model.t7 ] ; then
  5. th tools/rebuild_model.lua -i models/${1}/scale2.0x_model.t7 -o models/${1}/scale2.0x_model.t7 -backend cunn -model $2
  6. fi
  7. if [ -f models/${1}/noise1_model.t7 ] && [ ! -h models/${1}/noise1_model.t7 ]; then
  8. th tools/rebuild_model.lua -i models/${1}/noise1_model.t7 -o models/${1}/noise1_model.t7 -backend cunn -model $2
  9. fi
  10. if [ -f models/${1}/noise2_model.t7 ] && [ ! -h models/${1}/noise2_model.t7 ]; then
  11. th tools/rebuild_model.lua -i models/${1}/noise2_model.t7 -o models/${1}/noise2_model.t7 -backend cunn -model $2
  12. fi
  13. if [ -f models/${1}/noise3_model.t7 ] && [ ! -h models/${1}/noise3_model.t7 ]; then
  14. th tools/rebuild_model.lua -i models/${1}/noise3_model.t7 -o models/${1}/noise3_model.t7 -backend cunn -model $2
  15. fi
  16. }
  17. rebuild vgg_7/art vgg_7
  18. rebuild vgg_7/art_y vgg_7
  19. rebuild vgg_7/photo vgg_7
  20. rebuild vgg_7/ukbench vgg_7
  21. rebuild upconv_7/art upconv_7