1 files changed, 5 insertions, 0 deletions
diff --git a/Code/process_data.py b/Code/process_data.py
index 446de91..8375035 100644
--- a/Code/process_data.py
+++ b/Code/process_data.py
@@ -2,6 +2,7 @@ import numpy as np
 import getopt
 import sys
 from glob import glob
+import os
 
 import constants as c
 from utils import process_clip
@@ -64,6 +65,10 @@ def main():
             usage()
             sys.exit(2)
 
+    # set train frame dimensions
+    assert os.path.exists(c.TRAIN_DIR)
+    c.FULL_HEIGHT, c.FULL_WIDTH = c.get_train_frame_dims()
+
     ##
     # Process data for training
     ##