diff --git a/keras_wrapper/dataset.py b/keras_wrapper/dataset.py index 7a9bacc..32328ce 100644 --- a/keras_wrapper/dataset.py +++ b/keras_wrapper/dataset.py @@ -3037,7 +3037,7 @@ def preprocessImages(self, path_list, id, set_name, img_size, img_size_crop, use return data - def setTrainMean(self, mean_image, id, use_RGB=True, normalization=False): + def setTrainMean(self, mean_image, id, normalization=False): """ Loads a pre-calculated training mean image, 'mean_image' can either be: @@ -3046,7 +3046,6 @@ def setTrainMean(self, mean_image, id, use_RGB=True, normalization=False): - string with the path to the stored image. :param mean_image: - :param user_RGB: set to False for grayscale images :param normalization: :param id: identifier of the type of input whose train mean is being introduced. """