diff --git a/datasets/fake/generate_dataset.py b/datasets/fake/generate_dataset.py index a6c2294..eac391d 100644 --- a/datasets/fake/generate_dataset.py +++ b/datasets/fake/generate_dataset.py @@ -22,12 +22,12 @@ IMAGE_SIZE_SHAPE = (IMAGE_SIZE, IMAGE_SIZE) # Generate PRNUs and images of phones. # Is such `np.maximum` probabilistically correct with our theoretical method? See #19. -def randomImage(scale): +def randomGaussianImage(scale): return np.random.normal(loc = 0, scale = scale, size = IMAGE_SIZE_SHAPE) -imagesWithoutPrnu = [[randomImage(scale = 1) for _ in range(NUMBER_OF_IMAGES_PER_PHONE)] for phoneIndex in range(NUMBER_OF_PHONES)] +imagesWithoutPrnu = [[randomGaussianImage(scale = 1) for _ in range(NUMBER_OF_IMAGES_PER_PHONE)] for phoneIndex in range(NUMBER_OF_PHONES)] -prnus = [randomImage(scale = PRNU_FACTOR) for _ in range(NUMBER_OF_PHONES)] +prnus = [randomGaussianImage(scale = PRNU_FACTOR) for _ in range(NUMBER_OF_PHONES)] imagesWithPrnu = [[imageWithoutPrnu + prnus[phoneIndex] for imageWithoutPrnu in imagesWithoutPrnu[phoneIndex]] for phoneIndex in range(NUMBER_OF_PHONES)]