From aa9f73e8a0efc8dfffb8c90a2381e1e992cf8daf Mon Sep 17 00:00:00 2001
From: Benjamin Loison <12752145+Benjamin-Loison@users.noreply.github.com>
Date: Tue, 30 Apr 2024 03:49:45 +0200
Subject: [PATCH] Fix random seed to improve reproducibility

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

diff --git a/datasets/raise/split_and_compare_prnus_of_subgroups.py b/datasets/raise/split_and_compare_prnus_of_subgroups.py
index 81d72a6..126d79e 100755
--- a/datasets/raise/split_and_compare_prnus_of_subgroups.py
+++ b/datasets/raise/split_and_compare_prnus_of_subgroups.py
@@ -6,7 +6,7 @@ from tqdm import tqdm
 from utils import denoise, iterativeMean, getColorChannel, escapeFilePath, Color, mergeSingleColorChannelImagesAccordingToBayerFilter, rescaleRawImageForDenoiser, updateExtremes, saveNpArray
 import sys
 import os
-from random import shuffle
+import random
 
 sys.path.insert(0, '../../algorithms/distance/')
 
@@ -19,8 +19,9 @@ IMAGES_FOLDER = 'flat-field/NEF'
 setting = escapeFilePath(IMAGES_FOLDER) + f'_{DENOISER}'
 
 imagesFileNames = os.listdir(IMAGES_FOLDER)
+random.seed(0)
 # To not have a bias (chronological for instance) when split to make subgroups.
-shuffle(imagesFileNames)
+random.shuffle(imagesFileNames)
 
 numberOfImagesPerSubgroup = len(imagesFileNames) // NUMBER_OF_SUBGROUPS