From 4dd52aae90130327d3f4a6280886bdb7f0654afd Mon Sep 17 00:00:00 2001 From: Benjamin Loison Date: Wed, 3 Apr 2024 16:57:49 +0200 Subject: [PATCH] Use `plt.imsave` instead of `toPilImage` --- datasets/raise/split_and_compare_prnus_of_subgroups.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/datasets/raise/split_and_compare_prnus_of_subgroups.py b/datasets/raise/split_and_compare_prnus_of_subgroups.py index c3c2bac..f656b89 100644 --- a/datasets/raise/split_and_compare_prnus_of_subgroups.py +++ b/datasets/raise/split_and_compare_prnus_of_subgroups.py @@ -37,7 +37,7 @@ for subgroupIndex in range(NUMBER_OF_SUBGROUPS): # Not efficient mean computation. for numberOfImagesIndex, numberOfImages in enumerate(numberOfImagesThresholds): subgroupPrnuEstimateNpArray += [np.array(imagesPrnuEstimateNpArray[:numberOfImages]).mean(axis = 0)] - toPilImage(subgroupPrnuEstimateNpArray[-1]).save('prnu.png') + plt.imsave('prnu.png', subgroupPrnuEstimateNpArray[-1]) subgroupsPrnuEstimatesNpArray += [subgroupPrnuEstimateNpArray] rmss = []