From 1a910d2e2c49608297ad8737e44f98592e25a8e1 Mon Sep 17 00:00:00 2001 From: Benjamin Loison <12752145+Benjamin-Loison@users.noreply.github.com> Date: Mon, 13 May 2024 12:39:59 +0200 Subject: [PATCH] Add a try with `numpy.corrcoef` --- datasets/raise/attribute_source_camera.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/datasets/raise/attribute_source_camera.py b/datasets/raise/attribute_source_camera.py index 13d2f35..018d72e 100755 --- a/datasets/raise/attribute_source_camera.py +++ b/datasets/raise/attribute_source_camera.py @@ -138,7 +138,8 @@ for computeExtremes in tqdm(([True] if minColor is None or maxColor is None else #plt.imsave(f'{escapeFilePath(actualCamera)}_{cameraTestingImageIndex}.png', cameraTestingImagesNoise[actualCamera][cameraTestingImageIndex]) # Loop over each camera to compute closeness between the considered testing image noise and the estimated PRNUs of the various cameras. for camera in IMAGES_CAMERAS_FOLDER: - distance = rmsDiffNumpy(cameraTestingImagesNoise[actualCamera][cameraTestingImageIndex], camerasIterativeMean[camera].mean) + distance = abs(np.corrcoef(cameraTestingImagesNoise[actualCamera][cameraTestingImageIndex].flatten(), camerasIterativeMean[camera].mean.flatten()) - 1) + #distance = rmsDiffNumpy(cameraTestingImagesNoise[actualCamera][cameraTestingImageIndex], camerasIterativeMean[camera].mean) print(f'{cameraTestingImageIndex=} {camera=} {actualCamera=} {distance=}') if minimalDistance is None or distance < minimalDistance: minimalDistance = distance