From 612f4119fd99021dedec0013f9e069bba279ef8e Mon Sep 17 00:00:00 2001
From: Benjamin Loison <12752145+Benjamin-Loison@users.noreply.github.com>
Date: Fri, 3 May 2024 03:54:26 +0200
Subject: [PATCH] Determine the mean of camera images only on the training set

---
 datasets/raise/attribute_source_camera.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/datasets/raise/attribute_source_camera.py b/datasets/raise/attribute_source_camera.py
index b194a96..13697da 100755
--- a/datasets/raise/attribute_source_camera.py
+++ b/datasets/raise/attribute_source_camera.py
@@ -74,7 +74,7 @@ def getImagePrnuEstimateNpArray(singleColorChannelImages, multipleColorsImage, c
     return imagePrnuEstimateNpArray
 
 imagesCamerasFilePaths = {camera: [f'{IMAGES_CAMERAS_FOLDER[camera]}/{imagesCamerasFileName}' for imagesCamerasFileName in imagesCamerasFileNames[camera]] for camera in imagesCamerasFileNames}
-cameraColorMeans = {camera: getColorMeans(imagesCamerasFilePaths[camera], Color, minimalColorChannelCameraResolution) for camera in imagesCamerasFilePaths}
+cameraColorMeans = {camera: getColorMeans(imagesCamerasFilePaths[camera][:numberOfTrainingImages], Color, minimalColorChannelCameraResolution) for camera in imagesCamerasFilePaths}
 
 from utils import silentTqdm
 #tqdm = silentTqdm