Added training data existence checks

Author: matteodelv

bottlenecking.py

@@ -77,6 +77,7 @@ def train_top_model():
 
 
 if __name__ == '__main__':
+	u.checkDirs([s.train_data_dir, s.validation_data_dir, s.test_data_dir])
 	u.createDirIfNotExisting(s.plots_dir)
 	u.createDirIfNotExisting(s.results_dir)
 	save_bottlebeck_features()

model-evaluator.py

@@ -11,6 +11,9 @@
 # File paths
 log_path = os.path.join(s.results_dir, 'log_test_dataset.txt')
 
+# Check test data existence
+u.checkDirs([s.test_data_dir])
+
 # Load fine tuned model
 model = load_model(s.fine_tuned_model_path)
 print("Fine tuned model loaded...")

utils.py

@@ -44,3 +44,9 @@ def createDirIfNotExisting(path):
 	except OSError as e:
 		if e.errno != errno.EEXIST:
 			raise
+
+# Raise exception if training data is not found
+def checkDirs(paths):
+	for (i, path) in enumerate(paths):
+		if not os.path.isdir(path):
+			raise IOError(errno.ENOENT, os.strerror(errno.ENOENT), path)