Spaces:
Runtime error
Runtime error
File size: 477 Bytes
1f3f870 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
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)
|