diff --git a/datasets/raise/fft/verify_dots.py b/datasets/raise/fft/verify_dots.py new file mode 100644 index 0000000..c68a8d8 --- /dev/null +++ b/datasets/raise/fft/verify_dots.py @@ -0,0 +1,28 @@ +import os +from tqdm import tqdm +import numpy as np +import matplotlib.pyplot as plt +import sys + +sys.path.insert(0, '../') + +from utils import isARawImage, Color, getColorChannel, mergeSingleColorChannelImagesAccordingToBayerFilter, iterativeMean + +#images = [] +folder = '../rafael/230424' +meanImages = iterativeMean() + +for file in tqdm(os.listdir(folder)): + if isARawImage(file): + #print(file) + filePath = f'{folder}/{file}' + imageColorChannels = {color: getColorChannel(filePath, color) for color in Color} + imageMergedColorChannels = mergeSingleColorChannelImagesAccordingToBayerFilter(imageColorChannels) + #images += [imageMergedColorChannels] + meanImages.add(imageMergedColorChannels) + +#print(np.mean(images, axis = 0).shape) +print(meanImages.shape) + +plt.imshow() +plt.show() \ No newline at end of file