updates to tensorflow code and processing additionsHEADmaster

1 files changed, 24 insertions, 7 deletions

diff --git a/Code/constants.py b/Code/constants.py
index 761448b..5d5446b 100644
--- a/Code/constants.py
+++ b/Code/constants.py
@@ -42,18 +42,22 @@ def clear_dir(directory):
         except Exception as e:
             print(e)
 
+def get_process_frame_dims():
+    img_path = glob(os.path.join(PROCESS_DIR, '*'))[0]
+    img = imread(img_path, mode='RGB')
+    shape = np.shape(img)
+    return shape[0], shape[1]
+
 def get_test_frame_dims():
     img_path = glob(os.path.join(TEST_DIR, '*/*'))[0]
     img = imread(img_path, mode='RGB')
     shape = np.shape(img)
-
     return shape[0], shape[1]
 
 def get_train_frame_dims():
     img_path = glob(os.path.join(TRAIN_DIR, '*/*'))[0]
     img = imread(img_path, mode='RGB')
     shape = np.shape(img)
-
     return shape[0], shape[1]
 
 def set_test_dir(directory):
@@ -67,15 +71,28 @@ def set_test_dir(directory):
     TEST_DIR = directory
     FULL_HEIGHT, FULL_WIDTH = get_test_frame_dims()
 
+def set_process_dir(directory):
+    """
+    Edits all constants dependent on TEST_DIR.
+
+    @param directory: The new test directory.
+    """
+    global PROCESS_DIR, FULL_HEIGHT, FULL_WIDTH
+
+    TEST_DIR = directory
+    FULL_HEIGHT, FULL_WIDTH = get_process_frame_dims()
+
 # root directory for all data
 DATA_DIR = get_dir('../Data/')
 # directory of unprocessed training frames
-TRAIN_DIR = os.path.join(DATA_DIR, 'Ms_Pacman/Train/')
+TRAIN_DIR = os.path.join(DATA_DIR, 'WoodFlat/Train/')
 # directory of unprocessed test frames
-TEST_DIR = os.path.join(DATA_DIR, 'Ms_Pacman/Test/')
+TEST_DIR = os.path.join(DATA_DIR, 'WoodFlat/Test/')
+# directory of all the images we want to process
+PROCESS_DIR = os.path.join(DATA_DIR, 'WoodFlat/Process/')
 # Directory of processed training clips.
 # hidden so finder doesn't freeze w/ so many files. DON'T USE `ls` COMMAND ON THIS DIR!
-TRAIN_DIR_CLIPS = get_dir(os.path.join(DATA_DIR, '.Clips/'))
+TRAIN_DIR_CLIPS = get_dir(os.path.join(DATA_DIR, 'Clips/WoodClips/'))
 
 # For processing clips. l2 diff between frames must be greater than this
 MOVEMENT_THRESHOLD = 100
@@ -83,8 +100,8 @@ MOVEMENT_THRESHOLD = 100
 NUM_CLIPS = len(glob(TRAIN_DIR_CLIPS + '*'))
 
 # the height and width of the full frames to test on. Set in avg_runner.py or process_data.py main.
-FULL_HEIGHT = 210
-FULL_WIDTH = 160
+FULL_HEIGHT = 320
+FULL_WIDTH = 180
 # the height and width of the patches to train on
 TRAIN_HEIGHT = TRAIN_WIDTH = 32