diff --git a/datasets/raise/extract_noise.py b/datasets/raise/extract_noise.py index 359971f..cef5d8a 100755 --- a/datasets/raise/extract_noise.py +++ b/datasets/raise/extract_noise.py @@ -123,8 +123,8 @@ for color in Color: mean = None numberOfImagesInMean = 0 - for imageFileName in tqdm(imagesFileNames, 'Denoising images'): - treatImage(imageFileName) + for imageFileName in tqdm(imagesFileNames, f'Denoising images for color {color}'): + treatImage(imageFileName, color = color) npArrayFilePath = f'mean_{imagesFolderPathFileName}_{denoiser}_{color}.npy' with open(npArrayFilePath, 'wb') as f: