From ad2e609362fd33a3955d9f209e212415c476c94b Mon Sep 17 00:00:00 2001 From: Tae Young Kim Date: Thu, 12 Nov 2020 11:57:12 +0900 Subject: [PATCH] Change DepthDataset class __getitem__ function --- UtilityTest.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/UtilityTest.py b/UtilityTest.py index 087894d..5327dac 100644 --- a/UtilityTest.py +++ b/UtilityTest.py @@ -27,7 +27,7 @@ def __len__(self): def __getitem__(self, idx): - img_name = os.path.join(self.root_dir,os.listdir(self.root_dir)[idx]) + img_name = os.path.join(self.root_dir,sorted(os.listdir(self.root_dir))[idx]) image = (Image.open(img_name)) sample1={'image': image}