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

Enable computing specific colors

Browse Source 
For instance if have tested on one and do not want to waste its
computation.
This commit is contained in:
Benjamin Loison 2024-04-18 01:14:33 +02:00
parent 9365b58d1d
commit 29f3acc479
No known key found for this signature in database
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
datasets/raise/extract_noise.py
View File

@ -15,9 +15,9 @@ imagesFolderPathFileName = imagesFolderPath.replace('/', '_')
denoiser = 'bilateral'
raiseNotFlatFields = False
singleColorChannel = Color.GREEN_RIGHT
singleColorChannels = []
colors = Color if singleColorChannel is None else [singleColorChannel]
colors = singleColorChannels or Color
denoise = getattr(skimage.restoration, f'denoise_{denoiser}')