Skip to content

Commit f9da9f8

Browse files
committed
Minor changes to set the training mean image
1 parent a289b72 commit f9da9f8

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

keras_wrapper/dataset.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3037,7 +3037,7 @@ def preprocessImages(self, path_list, id, set_name, img_size, img_size_crop, use
30373037

30383038
return data
30393039

3040-
def setTrainMean(self, mean_image, id, use_RGB=True, normalization=False):
3040+
def setTrainMean(self, mean_image, id, normalization=False):
30413041
"""
30423042
Loads a pre-calculated training mean image, 'mean_image' can either be:
30433043
@@ -3046,7 +3046,6 @@ def setTrainMean(self, mean_image, id, use_RGB=True, normalization=False):
30463046
- string with the path to the stored image.
30473047
30483048
:param mean_image:
3049-
:param user_RGB: set to False for grayscale images
30503049
:param normalization:
30513050
:param id: identifier of the type of input whose train mean is being introduced.
30523051
"""

0 commit comments

Comments
 (0)