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