Files
Robust_image_source_identif…/datasets/raise/merge_single_color_channel_images_according_to_bayer_filter.py
Benjamin Loison c9f75286bf Make merge_single_color_channel_images_according_to_bayer_filter.py
virtual environment friendly
2024-04-30 03:20:44 +02:00

18 lines
586 B
Python
Executable File

#!/usr/bin/env python
from PIL import Image
from utils import Color, mergeSingleColorChannelImagesAccordingToBayerFilter
import matplotlib.pyplot as plt
import numpy as np
PREFIX = 'mean_rafael_arw_sky_mean_'
def getImageByColor(color):
filePath = PREFIX + f'{color}.npy'
image = np.load(filePath)
return image
singleColorChannelImages = {color: getImageByColor(color) for color in Color}
multipleColorsImage = mergeSingleColorChannelImagesAccordingToBayerFilter(singleColorChannelImages, progress = True)
plt.imsave(PREFIX + 'multiple_colors.png', multipleColorsImage)