From 856fc1973b0d63ef1841e37c3391dc49484e59a8 Mon Sep 17 00:00:00 2001
From: Matt Cooper <matthew_cooper@brown.edu>
Date: Fri, 12 Aug 2016 20:36:02 -0400
Subject: Made sure c.TEST_DIR exists before running

---
 Code/avg_runner.py | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'Code/avg_runner.py')

diff --git a/Code/avg_runner.py b/Code/avg_runner.py
index 5bf1e90..18bc592 100644
--- a/Code/avg_runner.py
+++ b/Code/avg_runner.py
@@ -1,6 +1,7 @@
 import tensorflow as tf
 import getopt
 import sys
+import os
 
 from utils import get_train_batch, get_test_batch
 import constants as c
@@ -158,6 +159,10 @@ def main():
         if opt == '--model_save_freq':
             c.MODEL_SAVE_FREQ = int(arg)
 
+    # set test frame dimensions
+    assert os.path.exists(c.TEST_DIR)
+    c.TEST_HEIGHT, c.TEST_WIDTH = c.get_test_frame_dims()
+
     ##
     # Init and run the predictor
     ##
-- 
cgit v1.2.3-70-g09d2