Benjamin_Loison/Robust_image_source_identification_on_modern_smartphones
1
0
Fork 0
Code Issues 68 Pull Requests Packages Projects Releases Wiki Activity

Fix random seed to improve reproducibility

Browse Source
This commit is contained in:
Benjamin Loison 2024-04-30 03:49:45 +02:00
parent 22fc4ce46b
commit aa9f73e8a0
No known key found for this signature in database
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
datasets/raise/split_and_compare_prnus_of_subgroups.py
View File

@ -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