Compare commits
2 Commits
5fa61f7ff8
...
60eceb575e
Author | SHA1 | Date | |
---|---|---|---|
60eceb575e | |||
b375acbb3a |
@ -5,10 +5,11 @@ from matplotlib import pyplot as plt
|
||||
def randomGaussianImage(scale, size):
|
||||
return np.random.normal(loc = 0, scale = scale, size = size)
|
||||
|
||||
def showImageWithMatplotlib(npArray, title = None):
|
||||
# `cmap` source: https://matplotlib.org/3.8.0/api/_as_gen/matplotlib.pyplot.imshow.html
|
||||
def showImageWithMatplotlib(npArray, title = None, cmap = 'viridis'):
|
||||
if title is not None:
|
||||
plt.title(title)
|
||||
plt.imshow(npArray)
|
||||
plt.imshow(npArray, cmap = cmap)
|
||||
plt.show()
|
||||
|
||||
def toPilImage(npArray):
|
||||
|
@ -62,5 +62,7 @@ for imageName in os.listdir(datasetPath):
|
||||
|
||||
cameraPrnuEstimateNpArray = np.array(imagesPrnuEstimateNpArray).mean(axis = 0)
|
||||
rms = rmsDiffNumpy(cameraPrnuEstimateNpArray, prnuNpArray, True)
|
||||
showImageWithMatplotlib(cameraPrnuEstimateNpArray, f'Camera PRNU estimate\nRMS with actual one: {rmsDiffNumpy(cameraPrnuEstimateNpArray, prnuNpArray):.4f} (normalized RMS: {rmsDiffNumpy(cameraPrnuEstimateNpArray, prnuNpArray, True):.4f})')
|
||||
title = f'Camera PRNU estimate\nRMS with actual one: {rmsDiffNumpy(cameraPrnuEstimateNpArray, prnuNpArray):.4f} (normalized RMS: {rmsDiffNumpy(cameraPrnuEstimateNpArray, prnuNpArray, True):.4f})'
|
||||
#showImageWithMatplotlib(cameraPrnuEstimateNpArray, title)
|
||||
|
||||
showImageWithMatplotlib(cameraPrnuEstimateNpArray, title, 'gray')
|
Loading…
Reference in New Issue
Block a user