From f181a3498c17178645197d012e0d8abcaf0095f2 Mon Sep 17 00:00:00 2001 From: Benjamin Loison Date: Tue, 26 Mar 2024 01:45:47 +0100 Subject: [PATCH] Correct `Image wuthout PRNU` rendering and precise that it is the first image --- datasets/fake/generate_dataset.py | 10 ++-------- 1 file changed, 2 insertions(+), 8 deletions(-) diff --git a/datasets/fake/generate_dataset.py b/datasets/fake/generate_dataset.py index 6646b17..3233cce 100644 --- a/datasets/fake/generate_dataset.py +++ b/datasets/fake/generate_dataset.py @@ -54,14 +54,8 @@ mainAxis = axes[MAIN_AXIS_ROW_INDEX] mainAxis[0].set_title('Actual PRNU') mainAxis[0].imshow(prnus[0]) -mainAxis[1].set_title('Image without PRNU') -mainAxis[1].imshow(imagesWithPrnu[0][0]) - -''' -imagesWithPrnuPil = [[toPilImage(imageWithPrnu) for imageWithPrnu in imagesWithPrnu[phoneIndex]] for phoneIndex in range(NUMBER_OF_PHONES)] -mainAxis[2].set_title(f'First image with PRNU\nRMS with image without PRNU: {round(rmsDiffNumpy(imagesWithPrnu[0][0], imagesWithoutPrnu[0][0]), 4)}') -mainAxis[2].imshow(imagesWithPrnu[0][0]) -''' +mainAxis[1].set_title('First image without PRNU') +mainAxis[1].imshow(imagesWithoutPrnu[0][0]) assert NUMBER_OF_IMAGES_PER_PHONE >= 10 ** (NUMBER_OF_ROWS - 1), 'Try to use more images than generated!'