From 9f2334ceea50d24ec7939d3992c38be2a54bbedf Mon Sep 17 00:00:00 2001
From: Benjamin Loison <12752145+Benjamin-Loison@users.noreply.github.com>
Date: Sun, 21 Apr 2024 19:54:04 +0200
Subject: [PATCH] Precise `imagesFolderPathFileName` depending on `sky` or
 `wall`

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

diff --git a/datasets/raise/extract_noise.py b/datasets/raise/extract_noise.py
index f8d169f..56dd937 100755
--- a/datasets/raise/extract_noise.py
+++ b/datasets/raise/extract_noise.py
@@ -33,9 +33,13 @@ if raiseNotFlatFields:
 
     imagesFileNames = [imageFileName for imageFileName in tqdm(imagesFileNames, 'Filtering images') if files[imageFileName]['Device'] == 'Nikon D7000' and Image.open(f'{imagesFolderPath}/png/{imageFileName}').size == (4946, 3278)]
 
-SKY = range(2_699, 2_807)
-WALL = range(2_807, 2_912)
-imagesFileNames = [f'DSC0{imageIndex}.ARW' for imageIndex in SKY]
+ranges = {
+    'sky': range(2_699, 2_807),
+    'wall': range(2_807, 2_912),
+}
+type_ = 'sky'
+imagesFileNames = [f'DSC0{imageIndex}.ARW' for imageIndex in ranges[type_]]
+imagesFolderPathFileName += f'_{type_}'
 
 minColor = None
 maxColor = None