diff --git a/datasets/raise/extract_noise.py b/datasets/raise/extract_noise.py index 74fa4d3..501463f 100755 --- a/datasets/raise/extract_noise.py +++ b/datasets/raise/extract_noise.py @@ -15,9 +15,9 @@ imagesFolderPathFileName = imagesFolderPath.replace('/', '_') denoiser = 'bilateral' raiseNotFlatFields = False -singleColorChannel = Color.GREEN_RIGHT +singleColorChannels = [] -colors = Color if singleColorChannel is None else [singleColorChannel] +colors = singleColorChannels or Color denoise = getattr(skimage.restoration, f'denoise_{denoiser}')