From c65b3642d8fb6ef615c6de8c331b963b299f413d Mon Sep 17 00:00:00 2001 From: Benjamin Loison <12752145+Benjamin-Loison@users.noreply.github.com> Date: Tue, 23 Apr 2024 05:01:22 +0200 Subject: [PATCH] Precise `multiple_colors.png` file name --- ...single_color_channel_images_according_to_bayer_filter.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 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 0cd94a3..2fc6600 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 @@ -4,8 +4,10 @@ from tqdm import tqdm import matplotlib.pyplot as plt import numpy as np +PREFIX = 'mean_rafael_arw_sky_mean_' + def getImageByColor(color): - filePath = f'means/mean_rafael_arw_bilateral_{color}.npy' + filePath = PREFIX + f'{color}.npy' image = np.load(filePath) return image @@ -39,4 +41,4 @@ for color in tqdm(Color, 'Color'): multipleColorsImage[newX, newY] = pixel -plt.imsave('multiple_colors.png', multipleColorsImage) \ No newline at end of file +plt.imsave(PREFIX + 'multiple_colors.png', multipleColorsImage)