diff --git a/datasets/raise/extract_noise.py b/datasets/raise/extract_noise.py index f7ce708..f99174e 100755 --- a/datasets/raise/extract_noise.py +++ b/datasets/raise/extract_noise.py @@ -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