From e080e841f57cf9ab20d66535a3c0930d3eacb625 Mon Sep 17 00:00:00 2001
From: Benjamin Loison <12752145+Benjamin-Loison@users.noreply.github.com>
Date: Wed, 17 Apr 2024 14:10:11 +0200
Subject: [PATCH] Add and use `raiseNotFlatFields`

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

diff --git a/datasets/raise/extract_noise.py b/datasets/raise/extract_noise.py
index 9f0183d..41971d0 100755
--- a/datasets/raise/extract_noise.py
+++ b/datasets/raise/extract_noise.py
@@ -14,15 +14,16 @@ imagesFolderPathFileName = imagesFolderPath.replace('/', '_')
 denoiser = 'wavelet'
 npArrayFilePath = f'mean_{imagesFolderPathFileName}_{denoiser}.npy'
 
+raiseNotFlatFields = False
+
 denoise = getattr(skimage.restoration, f'denoise_{denoiser}')
 
 mean = None
 numberOfImagesInMean = 0
 
-imagesFileNames = os.listdir(imagesFolderPath + '/png')
+imagesFileNames = os.listdir(imagesFolderPath + ('/png' if raiseNotFlatFields else ''))
 
-requiresRaiseFiltering = True
-if requiresRaiseFiltering:
+if raiseNotFlatFields:
     files = {}
 
     with open('RAISE_all.csv') as csvfile:
@@ -39,8 +40,11 @@ maxColor = None
 
 def treatImage(imageFileName, computeExtremes = False):
     global mean, numberOfImagesInMean, minColor, maxColor
-    imageFileName = imageFileName.replace('.png', '.NEF')
-    imageFilePath = f'{imagesFolderPath}/nef/{imageFileName}'
+    if raiseNotFlatFields:
+        imageFileName = imageFileName.replace('.png', '.NEF')
+        imageFilePath = f'{imagesFolderPath}/nef/{imageFileName}'
+    else:
+        imageFilePath = f'{imagesFolderPath}/{imageFileName}'
     if imageFileName.endswith('.NEF') or imageFileName.endswith('.ARW'):
         with rawpy.imread(imageFilePath) as raw:
             colorDesc = raw.color_desc.decode('ascii')