File size: 380 Bytes
6670ec8
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
import cv2
cv2.setNumThreads(1)
from os import path as osp
from basicsr.utils import scandir


def paths_from_folder(folder, key):
    gt_paths = list(scandir(folder))
    paths = []
    for idx in range(len(gt_paths)):
        gt_path = gt_paths[idx]
        gt_path = osp.join(folder, gt_path)
        paths.append(
            dict([(f'{key}_path', gt_path)]))
    return paths