Function get_diode_loader

dzoedepth/data/diode.py:121–123

(data_dir_root, batch_size=1, **kwargs)

def get_diode_loader(data_dir_root, batch_size=1, **kwargs):
    dataset = DIODE(data_dir_root)
    return DataLoader(dataset, batch_size, **kwargs)

# get_diode_loader(data_dir_root="datasets/diode/val/outdoor")