From 3fc0d64512cfc2fe7cef34afee26c85e2f026c50 Mon Sep 17 00:00:00 2001 From: Benjamin Loison Date: Thu, 18 Apr 2024 00:42:50 +0200 Subject: [PATCH] Add and use `getImageFileNameByColor` --- ...single_color_channel_images_according_to_bayer_filter.py | 6 +++++- datasets/raise/show_mean_noise.py | 2 +- 2 files changed, 6 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 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 diff --git a/datasets/raise/show_mean_noise.py b/datasets/raise/show_mean_noise.py index a9ff8b5..50d18f5 100644 --- a/datasets/raise/show_mean_noise.py +++ b/datasets/raise/show_mean_noise.py @@ -1,7 +1,7 @@ import numpy as np import matplotlib.pyplot as plt -fileName = 'mean_flat-field_nef_wavelet_blue' +fileName = 'mean_flat-field_nef_wavelet_green_bottom' npArray = np.load(f'{fileName}.npy') npArrayMin = npArray.min() print(f'{npArrayMin=}')