From e2406071d410a741fafd10ca510f15fe5582dbbd Mon Sep 17 00:00:00 2001
From: Benjamin Loison <12752145+Benjamin-Loison@users.noreply.github.com>
Date: Thu, 18 Apr 2024 01:16:41 +0200
Subject: [PATCH] Simplify specifying colors

---
 datasets/raise/extract_noise.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

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