diff --git a/hog_cnn.py b/hog_cnn.py index f88ead8..46ea226 100644 --- 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 index caf8695..92868d6 100644 --- 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 index 5ab29a3..5105d05 100644 --- 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/')