Benjamin_Loison/Robust_image_source_identification_on_modern_smartphones
1
0
Fork 0
Code Issues 68 Pull Requests Packages Projects Releases Wiki Activity

Show both RMS and RMS normalized

Browse Source
This commit is contained in:
Benjamin Loison 2024-03-29 01:47:01 +01:00
parent 3ccec5bbd0
commit 2f66e82f22
Signed by: Benjamin_Loison
SSH Key Fingerprint: SHA256:BtnEgYTlHdOg1u+RmYcDE0mnfz1rhv5dSbQ2gyxW8B8
2 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
algorithms/distance/rms_diff.py
View File

@ -14,5 +14,13 @@ def rmsDiffPil(im1, im2):
map(lambda h, i: h*(i**2), h, range(256))
) / (float(im1.size[0]) * im1.size[1]))
def rmsDiffNumpy(image0, image1):
def normalizeImage(image):
image = image - image.min()
image = image / image.max()
return image
def rmsDiffNumpy(image0, image1, normalize = False):
if normalize:
image0 = normalizeImage(image0)
image1 = normalizeImage(image1)
return np.sqrt(np.mean(np.square(image0 - image1)))

4
datasets/noise_free_test_images/estimate_prnu.py
View File

@ -58,6 +58,6 @@ for imageName in os.listdir(datasetPath):
imagesPrnuEstimateNpArray += [imagePrnuEstimateNpArray]
cameraPrnuEstimateNpArray = np.array(imagesPrnuEstimateNpArray).mean(axis = 0)
rms = rmsDiffNumpy(cameraPrnuEstimateNpArray, prnuNpArray)
showImageWithMatplotlib(cameraPrnuEstimateNpArray, f'Camera PRNU estimate\nRMS with actual one: {rms:.4f}')
#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})')