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)