Add and use getColorChannel

datasets/raise/split_and_compare_prnus_of_subgroups.py

@@ -4,7 +4,7 @@ from PIL import Image
 import numpy as np
 import matplotlib.pyplot as plt
 from tqdm import tqdm
-from utils import denoise, iterativeMean, escapeFilePath
+from utils import denoise, iterativeMean, getColorChannel, escapeFilePath, Color
 from skimage import img_as_float
 import sys
 import os
@@ -33,8 +33,8 @@ for subgroupImageIndex in tqdm(range(numberOfImagesPerSubgroup), 'Subgroup image
         imageIndex = (subgroupIndex * NUMBER_OF_SUBGROUPS) + subgroupImageIndex
         imageFileName = imagesFileNames[imageIndex]
         imageFilePath = f'{IMAGES_FOLDER}/{imageFileName}'
-        imagePil = Image.open(imageFilePath)
-        imageNpArray = img_as_float(np.array(imagePil))
+        imageNpArray = getColorChannel(imageFilePath, Color.GREEN_RIGHT)
+
         imagePrnuEstimateNpArray = imageNpArray - denoise(imageNpArray, DENOISER)
         subgroupIterativeMean = subgroupsIterativeMean[subgroupIndex]
         subgroupIterativeMean.add(imagePrnuEstimateNpArray)