Spaces:
Runtime error
Runtime error
import glob | |
import os | |
root_path = "images" | |
tasks = ["Deraining", "Dehazing", "Enhancement", "Deblurring", "Denoising"] | |
task_wise_image_paths = {} | |
for task in tasks: | |
task_path = os.path.join(root_path, task, "input") | |
image_paths = os.listdir(task_path) | |
image_paths = [os.path.join(task_path, image_path) for image_path in image_paths] | |
image_paths = [[i] for i in image_paths] | |
task_wise_image_paths.update({task: image_paths}) | |
print(task_wise_image_paths) | |