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:
@@ -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
|
||||
Reference in New Issue
Block a user