diff --git a/datasets/raise/utils.py b/datasets/raise/utils.py index a116edb..d2586c7 100644 --- a/datasets/raise/utils.py +++ b/datasets/raise/utils.py @@ -26,10 +26,10 @@ def denoise(imageNpArray, denoiserName): imageDenoisedNpArray = skImageRestorationDenoise(imageNpArray, weight=0.2) return imageDenoisedNpArray -class iterativeMean(): +class iterativeMean: mean = 0 numberOfElementsInMean = 0 - def addElement(self, element): + def add(self, element): self.mean = ((self.mean * self.numberOfElementsInMean) + element) / (self.numberOfElementsInMean + 1) self.numberOfElementsInMean += 1 \ No newline at end of file