ENSEEIHT/projet-long
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

change the order of asserts

Browse source 
Former-commit-id: 6cd4b68c997542cf0a9ca205120fd4f275e3cb24
This commit is contained in:
Gouvernathor 2022-04-06 13:45:19 +02:00 committed by GitHub
parent 8848edf8f3
commit 99e36df431
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
utils/data_loading.py
View file

@ -58,8 +58,8 @@ class BasicDataset(Dataset):
mask_file = list(self.masks_dir.glob(name + self.mask_suffix + '.*')) mask_file = list(self.masks_dir.glob(name + self.mask_suffix + '.*'))
img_file = list(self.images_dir.glob(name + '.*')) img_file = list(self.images_dir.glob(name + '.*'))
assert len(mask_file) == 1, f'Either no mask or multiple masks found for the ID {name}: {mask_file}'
assert len(img_file) == 1, f'Either no image or multiple images found for the ID {name}: {img_file}' assert len(img_file) == 1, f'Either no image or multiple images found for the ID {name}: {img_file}'
assert len(mask_file) == 1, f'Either no mask or multiple masks found for the ID {name}: {mask_file}'
mask = self.load(mask_file[0]) mask = self.load(mask_file[0])
img = self.load(img_file[0]) img = self.load(img_file[0])