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)