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

print assert error correctly

Browse source 
Former-commit-id: 8c7feff7cf740e788228839883a592c87097add2
This commit is contained in:
qslia 2022-04-10 07:54:48 +08:00
parent 182cbd48f9
commit 52ced2e757
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

@ -64,7 +64,7 @@ class BasicDataset(Dataset):
img = self.load(img_file[0]) img = self.load(img_file[0])
assert img.size == mask.size, \ assert img.size == mask.size, \
'Image and mask {name} should be the same size, but are {img.size} and {mask.size}' f'Image and mask {name} should be the same size, but are {img.size} and {mask.size}'
img = self.preprocess(img, self.scale, is_mask=False) img = self.preprocess(img, self.scale, is_mask=False)
mask = self.preprocess(mask, self.scale, is_mask=True) mask = self.preprocess(mask, self.scale, is_mask=True)