Use plt.imsave instead of toPilImage

This commit is contained in:
Benjamin Loison 2024-04-03 16:57:49 +02:00
parent d3af12ce3e
commit 4dd52aae90
Signed by: Benjamin_Loison
SSH Key Fingerprint: SHA256:BtnEgYTlHdOg1u+RmYcDE0mnfz1rhv5dSbQ2gyxW8B8

View File

@ -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 = []