diff --git a/datasets/raise/extract_noise.py b/datasets/raise/extract_noise.py index 501463f..f8d169f 100755 --- a/datasets/raise/extract_noise.py +++ b/datasets/raise/extract_noise.py @@ -15,9 +15,8 @@ imagesFolderPathFileName = imagesFolderPath.replace('/', '_') denoiser = 'bilateral' raiseNotFlatFields = False -singleColorChannels = [] - -colors = singleColorChannels or Color +# `[Color.RED, Color.GREEN_RIGHT, ...]` or `Color`. +colors = Color denoise = getattr(skimage.restoration, f'denoise_{denoiser}')