Rodrigo_Cobo commited on
Commit
3c20dc8
1 Parent(s): 59fba04

modified how to work with depth images in dataloader

Browse files
Files changed (1) hide show
  1. 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).convert('I')
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)