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!'