diff --git a/datasets/raise/utils.py b/datasets/raise/utils.py index 79fa11d..a116edb 100644 --- a/datasets/raise/utils.py +++ b/datasets/raise/utils.py @@ -24,4 +24,12 @@ def denoise(imageNpArray, denoiserName): imageDenoisedNpArray = skImageRestorationDenoise(imageNpArray, sigma_color=0.05, sigma_spatial=15) case 'tv_chambolle': imageDenoisedNpArray = skImageRestorationDenoise(imageNpArray, weight=0.2) - return imageDenoisedNpArray \ No newline at end of file + return imageDenoisedNpArray + +class iterativeMean(): + mean = 0 + numberOfElementsInMean = 0 + + def addElement(self, element): + self.mean = ((self.mean * self.numberOfElementsInMean) + element) / (self.numberOfElementsInMean + 1) + self.numberOfElementsInMean += 1 \ No newline at end of file