Rodrigo_Cobo
commited on
Commit
•
3c20dc8
1
Parent(s):
59fba04
modified how to work with depth images in dataloader
Browse files- dataloader.py +1 -2
dataloader.py
CHANGED
@@ -193,8 +193,7 @@ class TestingDataset(Dataset):
|
|
193 |
|
194 |
# foto produndidad
|
195 |
img_name = os.path.join(self.root_dir , self.files[idx][:-4] + "_d.png")
|
196 |
-
img = Image.open(img_name)
|
197 |
-
img = convert_I_to_L(img)
|
198 |
if (img.size[0] != self.imageDim or img.size[1] != self.imageDim):
|
199 |
img = img.resize((self.imageDim, self.imageDim))
|
200 |
x1_dep = transforms.ToTensor()(img)
|
|
|
193 |
|
194 |
# foto produndidad
|
195 |
img_name = os.path.join(self.root_dir , self.files[idx][:-4] + "_d.png")
|
196 |
+
img = Image.open(img_name)
|
|
|
197 |
if (img.size[0] != self.imageDim or img.size[1] != self.imageDim):
|
198 |
img = img.resize((self.imageDim, self.imageDim))
|
199 |
x1_dep = transforms.ToTensor()(img)
|