Laurent/REVA-QCAV
1
0
Fork 0
mirror of https://github.com/Laurent2916/REVA-QCAV.git synced 2024-11-09 23:12:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #187 from laclouis5/development

Browse source 
Fixed glob in dataset

Former-commit-id: 15df8266027d48ec1e78610338283c2a5dca7f7b
This commit is contained in:
milesial 2020-06-03 14:24:32 -07:00 committed by GitHub
parent 9fb406b393 4db3cb4620
commit aa9defe6a5
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
utils/dataset.py
View file

@ -43,8 +43,8 @@ class BasicDataset(Dataset):
def __getitem__(self, i): def __getitem__(self, i):
idx = self.ids[i] idx = self.ids[i]
mask_file = glob(self.masks_dir + idx + '*') mask_file = glob(self.masks_dir + idx + '.*')
img_file = glob(self.imgs_dir + idx + '*') img_file = glob(self.imgs_dir + idx + '.*')
assert len(mask_file) == 1, \ assert len(mask_file) == 1, \
f'Either no mask or multiple masks found for the ID {idx}: {mask_file}' f'Either no mask or multiple masks found for the ID {idx}: {mask_file}'