From ba5a1b742bdabc2bac272e728d00c338757de422 Mon Sep 17 00:00:00 2001 From: Benjamin Loison Date: Thu, 28 Mar 2024 22:20:40 +0100 Subject: [PATCH] #24: Make executions reproducible --- datasets/fake/generate_dataset.py | 2 ++ datasets/noise_free_test_images/estimate_prnu.py | 2 ++ 2 files changed, 4 insertions(+) diff --git a/datasets/fake/generate_dataset.py b/datasets/fake/generate_dataset.py index 3656497..008c813 100644 --- a/datasets/fake/generate_dataset.py +++ b/datasets/fake/generate_dataset.py @@ -26,6 +26,8 @@ PRNU_FACTOR = 0.1 IMAGE_SIZE_SHAPE = (IMAGE_SIZE, IMAGE_SIZE) +np.random.seed(0) + # Generate PRNUs and images of phones. imagesWithoutPrnu = [[randomGaussianImage(scale = 1, size = IMAGE_SIZE_SHAPE) for _ in range(NUMBER_OF_IMAGES_PER_PHONE)] for phoneIndex in range(NUMBER_OF_PHONES)] diff --git a/datasets/noise_free_test_images/estimate_prnu.py b/datasets/noise_free_test_images/estimate_prnu.py index bd07126..959cb46 100644 --- a/datasets/noise_free_test_images/estimate_prnu.py +++ b/datasets/noise_free_test_images/estimate_prnu.py @@ -15,6 +15,8 @@ datasetPath = 'no_noise_images' PRNU_FACTOR = 255 * 0.1 IMAGE_SIZE_SHAPE = (469, 704) +np.random.seed(0) + # Should first try with an image showing `PRNU`. prnu = randomGaussianImage(scale = PRNU_FACTOR, size = IMAGE_SIZE_SHAPE)