From 776fc205fe5e244cf5688cc96280edd7413d5865 Mon Sep 17 00:00:00 2001 From: Sebastiano Di Paola Date: Thu, 1 Feb 2018 22:20:26 +0100 Subject: [PATCH] Update Pipeline.py Allow also an absolute path to be specified as output directory --- Augmentor/Pipeline.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/Augmentor/Pipeline.py b/Augmentor/Pipeline.py index f2f6725..f49e943 100644 --- a/Augmentor/Pipeline.py +++ b/Augmentor/Pipeline.py @@ -121,9 +121,12 @@ def _populate(self, source_directory, output_directory, ground_truth_directory, if not os.path.exists(ground_truth_directory): raise IOError("The ground truth source directory you specified does not exist.") - # Get absolute path for output - abs_output_directory = os.path.join(source_directory, output_directory) - + # Allow to specity an absolute path for output directory + if(not os.path.isabs(output_directory)): + abs_output_directory = os.path.join(source_directory, output_directory) + else: + abs_output_directory = output_directory + # Scan the directory that user supplied. self.augmentor_images, self.class_labels = scan(source_directory, abs_output_directory)