Made video demo repeat on video loop.

Author: Iainmon

demos/video/style-transfer/export_and_run_model.sh

@@ -1,8 +1,12 @@
 
 
+MODEL_NAME=$1
 
-MODEL_NAME="oil_h_bt4_5e11_ep4_epoch_0_batch_id_18000"
+# MODEL_NAME="nature_oil_painting_ep4_bt4_sw1e10_cw_1e5_flash_epoch_1_batch_id_8000"
+# MODEL_NAME="nature_oil_painting_ep4_bt4_sw5e9_cw_1e5_flash_epoch_0_batch_id_12000"
+
+python3 neural_style.py export --model saved_models/${MODEL_NAME}.model --accel \
+    || python3 neural_style.py export --model saved_models/${MODEL_NAME}.pth --accel
 
-python3 neural_style.py export --model saved_models/${MODEL_NAME}.pth --accel
 
 python3 style_transfer_test.py --model-file=models/exports/cpu/${MODEL_NAME}_float16.pt --input-video-file=videos/deer.mp4 --show-output