diff --git a/datasets/raise/extract_noise.py b/datasets/raise/extract_noise.py index f8d169f..56dd937 100755 --- a/datasets/raise/extract_noise.py +++ b/datasets/raise/extract_noise.py @@ -33,9 +33,13 @@ 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)] -SKY = range(2_699, 2_807) -WALL = range(2_807, 2_912) -imagesFileNames = [f'DSC0{imageIndex}.ARW' for imageIndex in SKY] +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_}' minColor = None maxColor = None