From 81e571f1701a9df3752a6ab1ee0a1e023c22cc02 Mon Sep 17 00:00:00 2001 From: Martin Vejbora Date: Sun, 19 Mar 2023 00:30:07 +0100 Subject: [PATCH] Fix #8, prepending data_root path twice when loading image --- diode.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/diode.py b/diode.py index 7f56c9b..b30710e 100644 --- a/diode.py +++ b/diode.py @@ -112,7 +112,7 @@ def __getitem__(self, index): de_fname = osp.join(self.data_root, '{}_depth.npy'.format(im)) de_mask_fname = osp.join(self.data_root, '{}_depth_mask.npy'.format(im)) - im = np.array(Image.open(osp.join(self.data_root, im_fname))) + im = np.array(Image.open(im_fname)) de = np.load(de_fname).squeeze() de_mask = np.load(de_mask_fname) return im, de, de_mask