2 Commits
+3 -2
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: