Skip to content

Commit

Permalink
Cosmetic change committed before fetch and merge.
Browse files Browse the repository at this point in the history
  • Loading branch information
mdbloice committed May 11, 2018
1 parent 1620421 commit a987451
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion Augmentor/Pipeline.py
Original file line number Diff line number Diff line change
Expand Up @@ -179,7 +179,7 @@ def _check_images(self, abs_output_directory):
sys.stdout.write("Initialised with %s image(s) found.\n" % len(self.augmentor_images))
sys.stdout.write("Output directory set to %s." % abs_output_directory)

def _execute(self, augmentor_image, save_to_disk=True, multi_threaded=False):
def _execute(self, augmentor_image, save_to_disk=True, multi_threaded=True):
"""
Private method. Used to pass an image through the current pipeline,
and return the augmented image.
Expand Down Expand Up @@ -407,6 +407,7 @@ def image_generator(self):
yield self._execute(self.augmentor_images[im_index], save_to_disk=False), \
self.augmentor_images[im_index].class_label_int

# TODO: Fix: scaled=True results in an error.
def keras_generator(self, batch_size, scaled=True, image_data_format="channels_last"):
"""
Returns an image generator that will sample from the current pipeline
Expand Down

0 comments on commit a987451

Please sign in to comment.