From 6e7e06b85c6e28605e4e559730eebf14ad3bb86a Mon Sep 17 00:00:00 2001 From: Benjamin Loison <12752145+Benjamin-Loison@users.noreply.github.com> Date: Wed, 15 May 2024 19:57:56 +0200 Subject: [PATCH] Clean `attribute_source_camera.py` --- datasets/raise/attribute_source_camera.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/datasets/raise/attribute_source_camera.py b/datasets/raise/attribute_source_camera.py index 3e9b934..f8408c3 100755 --- a/datasets/raise/attribute_source_camera.py +++ b/datasets/raise/attribute_source_camera.py @@ -141,7 +141,8 @@ for computeExtremes in tqdm(([True] if minColor is None or maxColor is None else for cameraTestingImageIndex in tqdm(range(numberOfTestingImages), 'Camera testing image index'): print(f'{camera=} {numberOfTrainingImages + cameraTestingImageIndex=}') - singleColorChannelImages = singleColorChannelTestingImages[camera][cameraTestingImageIndex]#getSingleColorChannelImages(camera, numberOfTrainingImages + cameraTestingImageIndex) + #singleColorChannelImages = singleColorChannelTestingImages[camera][cameraTestingImageIndex] + singleColorChannelImages = getSingleColorChannelImages(camera, numberOfTrainingImages + cameraTestingImageIndex) multipleColorsImage = getMultipleColorsImage(singleColorChannelImages) imagePrnuEstimateNpArray = getImagePrnuEstimateNpArray(singleColorChannelImages, multipleColorsImage, camera) @@ -184,7 +185,8 @@ for computeExtremes in tqdm(([True] if minColor is None or maxColor is None else if DENOISER != Denoiser.MEAN: cameraTestingImageNoise = cameraTestingImagesNoise[actualCamera][cameraTestingImageIndex] else: - singleColorChannelImages = singleColorChannelTestingImages[camera][cameraTestingImageIndex]#getSingleColorChannelImages(camera, numberOfTrainingImages + cameraTestingImageIndex) + #singleColorChannelImages = singleColorChannelTestingImages[camera][cameraTestingImageIndex] + singleColorChannelImages = getSingleColorChannelImages(camera, numberOfTrainingImages + cameraTestingImageIndex) multipleColorsImage = getMultipleColorsImage(singleColorChannelImages) cameraTestingImageNoise = getImagePrnuEstimateNpArray(singleColorChannelImages, multipleColorsImage, camera)