Compare commits

..

2 Commits

Author SHA1 Message Date
Benjamin Loison
8748e648b5
Remove no more relevant commented code for a test 2024-04-06 00:06:13 +02:00
Benjamin Loison
15c41c0463
Add and use imageDenoisedNpArray to ease using other denoisers 2024-04-06 00:03:50 +02:00

View File

@ -13,14 +13,15 @@ npArrayFilePath = 'mean.npy'
mean = None
numberOfImagesInMean = 0
imagesFileNames = os.listdir(imagesFolderPath)#[:2]
imagesFileNames = os.listdir(imagesFolderPath)
def treatImage(imageFileName):
global mean, numberOfImagesInMean
imageFilePath = f'{imagesFolderPath}/{imageFileName}'
imagePil = Image.open(imageFilePath)
imageNpArray = img_as_float(np.array(imagePil))
imageNoiseNpArray = imageNpArray - denoise_tv_chambolle(imageNpArray, weight=0.2, channel_axis=-1)
imageDenoisedNpArray = denoise_tv_chambolle(imageNpArray, weight=0.2, channel_axis=-1)
imageNoiseNpArray = imageNpArray - imageDenoisedNpArray
if mean is None:
mean = imageNoiseNpArray
else: