From c287d5f0ef571dfbbde6839052a2f1cfc8d697b8 Mon Sep 17 00:00:00 2001 From: Benjamin Loison Date: Fri, 22 Mar 2024 11:39:51 +0100 Subject: [PATCH] Add an use `toPilImage` --- datasets/fake/generate_dataset.py | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/datasets/fake/generate_dataset.py b/datasets/fake/generate_dataset.py index 8a46490..f1a90c8 100644 --- a/datasets/fake/generate_dataset.py +++ b/datasets/fake/generate_dataset.py @@ -1,5 +1,6 @@ import numpy as np from matplotlib import pyplot as plt +from PIL import Image import sys sys.path.insert(0, '../../algorithms/distance/') @@ -9,16 +10,20 @@ from rmsdiff import rmsdiff IMAGE_SIZE = 64 def randomImage(scale): - # Is `np.clip` necessary? - return np.round(np.random.normal(loc = 0, scale = scale, size = (IMAGE_SIZE, IMAGE_SIZE))) + # Is `np.clip` necessary? See `toPilImage`. + return np.random.normal(loc = 0, scale = scale, size = (IMAGE_SIZE, IMAGE_SIZE)) prnu = randomImage(scale = 1) -plt.imshow(prnu) -plt.show() - images = [randomImage(scale = 10) + prnu for _ in range(10)] +def toPilImage(npArray): + nonNegativeArray = npArray - np.min(npArray) + nonNegativeArray = np.round(255 * nonNegativeArray / np.max(nonNegativeArray)) + return Image.fromarray(np.uint8(nonNegativeArray)) + +toPilImage(prnu).show() + for image in images: print(rmsdiff(image, prnu)) print(rmsdiff(contextAdaptiveInterpolator(image), prnu)) \ No newline at end of file