From a9a29ff9dd940d4099b5b7303290d7f1597b56c2 Mon Sep 17 00:00:00 2001 From: victorjoos Date: Thu, 16 May 2019 16:11:44 +0200 Subject: [PATCH] Fix dtype propagation --- keras_preprocessing/image/image_data_generator.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/keras_preprocessing/image/image_data_generator.py b/keras_preprocessing/image/image_data_generator.py index 5c926eb1..73b2bb1a 100644 --- a/keras_preprocessing/image/image_data_generator.py +++ b/keras_preprocessing/image/image_data_generator.py @@ -429,7 +429,8 @@ def flow(self, save_to_dir=save_to_dir, save_prefix=save_prefix, save_format=save_format, - subset=subset + subset=subset, + dtype=self.dtype ) def flow_from_directory(self, @@ -537,7 +538,8 @@ class subdirectories (default: False). save_format=save_format, follow_links=follow_links, subset=subset, - interpolation=interpolation + interpolation=interpolation, + dtype=self.dtype ) def flow_from_dataframe(self, @@ -680,7 +682,8 @@ def flow_from_dataframe(self, save_format=save_format, subset=subset, interpolation=interpolation, - validate_filenames=validate_filenames + validate_filenames=validate_filenames, + dtype=self.dtype ) def standardize(self, x):