From 29f3acc479638d2d0269fd82ec63ec738073f44b Mon Sep 17 00:00:00 2001
From: Benjamin Loison <12752145+Benjamin-Loison@users.noreply.github.com>
Date: Thu, 18 Apr 2024 01:14:33 +0200
Subject: [PATCH] Enable computing specific colors

For instance if have tested on one and do not want to waste its
computation.
---
 datasets/raise/extract_noise.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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}')