runner.py

from glob import glob
from time import sleep
from os import path, rename
from re import sub
from sys import stdout
from subprocess import call

PYTORCH_STYLE_TRANSFER_CWD = path.join(
  path.dirname(__file__),
  'PyTorch-Style-Transfer',
  'experiments'
)

def newstate(filepath, state):
  return sub(r'\.[^\.]+$', '.' + state, filepath)

def process(filepath):
  processing = newstate(filepath, "processing")
  rename(filepath, processing)

  style_path = path.abspath(path.join(processing, "style.jpg"))
  subject_path = path.abspath(path.join(processing, "subject.jpg"))
  output_path = path.abspath(path.join(processing, "output.jpg"))
  stdout_path = path.join(processing, "stdout.txt")

  with open(stdout_path, "a") as f:
    f.write("Status: starting\n")

  cmd = "python -u main.py optim " \
    "--style-image {0} --content-image {1} --output-image {2} " \
    "--iters 200 --log-interval 1 --style-weight 10" \
    "".format(style_path, subject_path, output_path)
  with open(stdout_path, 'a') as f:
    call(cmd.split(), stdout=f, bufsize=1, cwd=PYTORCH_STYLE_TRANSFER_CWD)
    f.write("Status: done\n")

count = 0
while(True):
  print "Checking for requests... ", count
  count += 1
  stdout.flush()
  files = list(reversed(sorted(glob("images/*.request"))))
  for file in files:
    print "request: ", path.basename(file)
    stdout.flush()
  if len(files) > 0:
    print "* Processing: ", files[0]
    stdout.flush()
    process(files[0])
  sleep(1)
  
  print
  stdout.flush()