#!/bin/env python

import os
import glob
import sys

os.chdir("/home/pux/p/nst_vid")
for filename in sorted(glob.glob('image-*.jpg')):
    num = filename[6:-4]
    print(filename)
    print(num)
    os.chdir("/home/pux/src/neural-style-pt")

    os.system("python neural_style.py -style_image /home/pux/p/nst_vid/style.jpg -content_image /home/pux/p/nst_vid/image-" + str(num) + ".jpg -output_image ~/p/nst_vid/generated-" + str(num) + ".jpg -image_size 384 -style_weight 1000 -content_weight 5 -cudnn_autotune -tv_weight 0.00085 -num_iterations 1000 -print_iter 50 -init image -optimizer lbfgs -learning_rate 1.0 -lbfgs_num_correction 100 -style_scale 1.0 -pooling max -backend cudnn -seed 100 -content_layers relu1_1,relu2_1,relu3_1,relu4_1,relu4_2,relu5_1 -style_layers relu3_1,relu4_1,relu4_2,relu5_1 -multidevice_strategy 4,7,29")

    os.system("python opticalflow.py --input_1 /home/pux/p/nst_vid/image-" + str(num) + ".jpg --input_1_styled /home/pux/p/nst_vid/generated-" + str(num) + ".jpg --input_2 /home/pux/p/nst_vid/image-" + str(int(num)+1).zfill(5) + ".jpg --output /home/pux/p/nst_vid/blended-" + str(num) + ".jpg --alpha 0.55 --flowblend False")