Renamed directory where the model is saved.

JTKBowers · Sep 29, 2016 · dc861da · dc861da
1 parent 2cdaf1d
commit dc861da
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/hog_cnn.py b/hog_cnn.py
@@ -134,7 +134,7 @@ def bbox_correct(bbox, example_bboxes):
     nn_im_h = 160
     with tf.Session() as sess:
         model = Model.BooleanModel(sess)
-        model.load('out/', nn_im_w, nn_im_h)
+        model.load('saved_model/', nn_im_w, nn_im_h)
 
         image_count = 0
         HOG_TP_count = 0

diff --git a/test_person_classification.py b/test_person_classification.py
@@ -42,7 +42,7 @@
     confusion_matrices = []
     with tf.Session() as sess:
         model = Model.BooleanModel(sess)
-        model.load('out/', nn_im_w, nn_im_h)
+        model.load('saved_model/', nn_im_w, nn_im_h)
 
         print("Testing...")
         test_accuracy, confusion_matrix = model.test(combined_dataset.test.iter_people())

diff --git a/train_person_classification.py b/train_person_classification.py
@@ -58,4 +58,4 @@ def train(self, person_iter):
         print("Training...")
         model.train(combined_dataset.train.iter_people())
 
-        model.save('out/')
+        model.save('saved_model/')