From c85ee573ed5a33c7741c61ba9fcce0ad64718d34 Mon Sep 17 00:00:00 2001 From: Matt Cooper Date: Sun, 21 Aug 2016 15:33:05 -0400 Subject: testing a bug where test inputs and GTs are out of order --- Code/process_data.py | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'Code/process_data.py') 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 ## -- cgit v1.2.3-70-g09d2