Move denoise from extract_noise.py to utils.py

See next commit leveraging it in `split_and_compare_prnus_of_subgroups.py`
This commit is contained in:
2024-04-26 13:57:18 +02:00
parent 6a93b04686
commit 9d2903af60
2 changed files with 23 additions and 18 deletions

View File

@@ -1,4 +1,5 @@
from enum import Enum, auto
import skimage.restoration
class Color(Enum):
RED = auto()
@@ -8,3 +9,19 @@ class Color(Enum):
def __str__(self):
return self.name.lower()
# Among:
# - `wavelet`
# - `bilateral`
# - `tv_chambolle`
def denoise(imageNpArray, denoiserName):
skImageRestorationDenoise = getattr(skimage.restoration, f'denoise_{denoiserName}')
match denoiserName:
case 'wavelet':
imageDenoisedNpArray = skImageRestorationDenoise(imageNpArray, rescale_sigma=True)
case 'bilateral':
imageDenoisedNpArray = skImageRestorationDenoise(imageNpArray, sigma_color=0.05, sigma_spatial=15)
case 'tv_chambolle':
imageDenoisedNpArray = skImageRestorationDenoise(imageNpArray, weight=0.2)
return imageDenoisedNpArray