Benjamin_Loison/Robust_image_source_identification_on_modern_smartphones
1
0
Fork 0
Code Issues 68 Pull Requests Packages Projects Releases Wiki Activity

Simplify specifying colors

Browse Source
This commit is contained in:
Benjamin Loison 2024-04-18 01:16:41 +02:00
parent 29f3acc479
commit e2406071d4
No known key found for this signature in database
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
datasets/raise/extract_noise.py
View File

@ -15,9 +15,8 @@ imagesFolderPathFileName = imagesFolderPath.replace('/', '_')
denoiser = 'bilateral'
raiseNotFlatFields = False
singleColorChannels = []
colors = singleColorChannels or Color
# `[Color.RED, Color.GREEN_RIGHT, ...]` or `Color`.
colors = Color
denoise = getattr(skimage.restoration, f'denoise_{denoiser}')