From 0b52e781e38cca81cdf2fc98559392f444c8ed71 Mon Sep 17 00:00:00 2001 From: Benjamin Loison Date: Thu, 18 Apr 2024 01:00:41 +0200 Subject: [PATCH] First try to merge multiple single color channel images --- ...hannel_images_according_to_bayer_filter.py | 23 +++++++++++-------- 1 file changed, 13 insertions(+), 10 deletions(-) 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 fa16348..7557e1b 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,31 +1,34 @@ from PIL import Image from utils import Color +from tqdm import tqdm def getImageByColor(color): - return Image.open(f'mean_flat-field_nef_wavelet_{color}.png') + return Image.open(f'means/mean_flat-field_nef_wavelet_{color}.png') color = Color.BLUE image = getImageByColor(color) -height, width = image.size +width, height= image.size multipleColorsImage = Image.new('RGB', [dimension * 2 for dimension in image.size]) -for color in Color: +for color in tqdm(Color, 'Color'): colorImage = getImageByColor(color) - for y in range(height): + for y in tqdm(range(height), 'Height'): for x in range(width): - pixel = colorImage.getpixel(x, y) + pixel = colorImage.getpixel((x, y)) ''' RG GB ''' newX, newY = { - Color.RED: 2 * x, 2 * y, - Color.GREEN_RIGHT: 2 * (x - 1), 2 * y, - Color.GREEN_BOTTOM: 2 * x, 2 * (y - 1), - Color.BLUE: 2 * (x - 1), 2 * (y - 1), + Color.RED: (x, y), + Color.GREEN_RIGHT: (x - 1, y), + Color.GREEN_BOTTOM: (x, y - 1), + Color.BLUE: (x - 1, y - 1), }[color] + newX *= 2 + newY *= 2 - multipleColorsImage.putpixel((x, y), pixel) + multipleColorsImage.putpixel((newX, newY), pixel) multipleColorsImage.save('multipleColors.png') \ No newline at end of file