diff --git a/datasets/raise/benchmark_load_part_of_images.py b/datasets/raise/benchmark_load_part_of_images.py index 18d3f54..e50efc2 100755 --- a/datasets/raise/benchmark_load_part_of_images.py +++ b/datasets/raise/benchmark_load_part_of_images.py @@ -7,8 +7,10 @@ import numpy as np from enum import Enum, auto IMAGES_CAMERAS_FOLDER = { - 'RAISE': 'flat-field/nef', - 'Rafael 23/04/24': 'rafael/230424', + #'RAISE': 'flat-field/nef', + #'Rafael 23/04/24': 'rafael/230424', + 'RAISE not flat-field D7000': 'not_flat-field/d7000/nef', + 'RAISE not flat-field D90': 'not_flat-field/d90/nef', } class Operation(Enum): @@ -44,4 +46,4 @@ for camera in tqdm(IMAGES_CAMERAS_FOLDER, 'Camera'): if OPERATION == Operation.LOAD_NPY: numpyFilePath = getNumpyFilePath(imageFilePath, color) rawColorChannel = np.load(numpyFilePath, mmap_mode = 'r') - print(color, rawColorChannel[:RESOLUTION].mean()) \ No newline at end of file + print(color, rawColorChannel[:RESOLUTION].mean())