Use plt.imsave instead of toPilImage

This commit is contained in:
2024-04-03 16:57:49 +02:00
parent d3af12ce3e
commit 4dd52aae90

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