diff options
Diffstat (limited to 'Code/avg_runner.py')
| -rw-r--r-- | Code/avg_runner.py | 5 |
1 files changed, 5 insertions, 0 deletions
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 ## |