diff --git a/datasets/raise/utils.py b/datasets/raise/utils.py index cb52932..c5043c6 100644 --- a/datasets/raise/utils.py +++ b/datasets/raise/utils.py @@ -145,7 +145,7 @@ def getColorMeans(imagesFileNames, colors, singleColorChannelCropResolution = No colorMeans = {} for color in colors: colorIterativeMean = iterativeMean() - for imageFileName in tqdm(imagesFileNames, f'Computing mean of {color} colored images'): + for imageFileName in tqdm(imagesFileNames, f'Computing mean of {color.replace("_", " ")} colored images'): imageNpArray = getImageNpArray(imageFileName, False, color, Denoiser.MEAN) if singleColorChannelCropResolution is not None: imageNpArray = getImageCrop(imageNpArray, singleColorChannelCropResolution)