Add verify_dots.py
This commit is contained in:
parent
fbd7cb01e1
commit
38eca8e65c
28
datasets/raise/fft/verify_dots.py
Normal file
28
datasets/raise/fft/verify_dots.py
Normal file
@ -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()
|
Loading…
x
Reference in New Issue
Block a user