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

Correct Rafael images filtering

Browse Source
This commit is contained in:
Benjamin Loison 2024-04-23 04:37:46 +02:00
parent e3bbb2c63d
commit 809aa7fb43
Signed by: Benjamin_Loison
SSH Key Fingerprint: SHA256:BtnEgYTlHdOg1u+RmYcDE0mnfz1rhv5dSbQ2gyxW8B8
1 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
datasets/raise/extract_noise.py
View File

@ -37,13 +37,18 @@ if raiseNotFlatFields:
imagesFileNames = [imageFileName for imageFileName in tqdm(imagesFileNames, 'Filtering images') if files[imageFileName]['Device'] == 'Nikon D7000' and Image.open(f'{imagesFolderPath}/png/{imageFileName}').size == (4946, 3278)]
ranges = {
'sky': range(2_699, 2_807),
'wall': range(2_807, 2_912),
}
type_ = 'sky'
imagesFileNames = [f'DSC0{imageIndex}.ARW' for imageIndex in ranges[type_]]
imagesFolderPathFileName += f'_{type_}'
# Among:
# - `None`
# - `"sky"`
# - `"wall"`
type_ = None
if type_ is not None:
ranges = {
'sky': range(2_699, 2_807),
'wall': range(2_807, 2_912),
}
imagesFileNames = [f'DSC0{imageIndex}.ARW' for imageIndex in ranges[type_]]
imagesFolderPathFileName += f'_{type_}'
minColor = None
maxColor = None