From be91a07dd5544a424de95247c46480e6c8607f2d Mon Sep 17 00:00:00 2001 From: Benjamin Loison Date: Thu, 18 Apr 2024 00:43:14 +0200 Subject: [PATCH] Add and use `getImageFileNameByColor` --- ...single_color_channel_images_according_to_bayer_filter.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/datasets/raise/merge_single_color_channel_images_according_to_bayer_filter.py b/datasets/raise/merge_single_color_channel_images_according_to_bayer_filter.py index c5e76bd..b8b6694 100644 --- a/datasets/raise/merge_single_color_channel_images_according_to_bayer_filter.py +++ b/datasets/raise/merge_single_color_channel_images_according_to_bayer_filter.py @@ -1,11 +1,15 @@ from PIL import Image from utils import Color +def getImageFileNameByColor(color): + return f'mean_flat-field_nef_wavelet_{color}.png' + color = Color.BLUE -image = Image.open(f'mean_flat-field_nef_wavelet_{color}.png') +image = Image.open(getImageFileNameByColor(color)) #print(image.size) newImage = Image.new('RGB', [dimension * 2 for dimension in image.size]) #print(newImage.size) for color in Color: + getImageFileNameByColor(color) \ No newline at end of file