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)