From 92ede944c76bb1e18a311b89da39152267836ac5 Mon Sep 17 00:00:00 2001 From: Benjamin Loison Date: Fri, 22 Mar 2024 11:42:47 +0100 Subject: [PATCH] Not satisfying due to zero offset --- datasets/fake/generate_dataset.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/datasets/fake/generate_dataset.py b/datasets/fake/generate_dataset.py index f1a90c8..d06b093 100644 --- a/datasets/fake/generate_dataset.py +++ b/datasets/fake/generate_dataset.py @@ -16,10 +16,11 @@ def randomImage(scale): prnu = randomImage(scale = 1) images = [randomImage(scale = 10) + prnu for _ in range(10)] +allImages = [prnu] + images def toPilImage(npArray): - nonNegativeArray = npArray - np.min(npArray) - nonNegativeArray = np.round(255 * nonNegativeArray / np.max(nonNegativeArray)) + nonNegativeArray = npArray - np.min(allImages) + nonNegativeArray = np.round(255 * nonNegativeArray / np.max(allImages)) return Image.fromarray(np.uint8(nonNegativeArray)) toPilImage(prnu).show()