Add and use getImageColorChannel

2024-05-23 14:48:11 +02:00
parent 3c983b0690
commit cc05f5c210
1 changed files with 5 additions and 2 deletions
--- a/datasets/raise/extract_4_color_channel_images_from_raw.py
+++ b/datasets/raise/extract_4_color_channel_images_from_raw.py
@@ -11,8 +11,11 @@ INTERESTING_POSITION = np.array(ORIGINAL_INTERESTING_POSITION) // 2
 def getImageFilePath(imageIndex):
    return f'../rafael/240424/photos/DSC0{imageIndex}.ARW'

-imageFilePath = getImageFilePath(IMAGE_INDEX)
-colorChannel = getColorChannel(imageFilePath, Color.RED)
+def getImageColorChannel(imageIndex, color):
+    imageFilePath = getImageFilePath(imageIndex)
+    return getColorChannel(imageFilePath, Color.RED)
+
+colorChannel = getImageColorChannel(IMAGE_INDEX, Color.RED)
 print(colorChannel.shape)

 def crop(image, interestingPosition, yRange, xRange):