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

Merge pull request #348 from pleaseRedo/master

Browse source 
Update utils.py. minor fix

Former-commit-id: 1c736851c43aceb70f49f6e444f1e22a377ec59d
This commit is contained in:
milesial 2022-03-31 21:37:48 +02:00 committed by GitHub
parent 2ca43802cc eb95dab01e
commit c1c7c952aa
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/utils.py
View file

@ -9,7 +9,7 @@ def plot_img_and_mask(img, mask):
if classes > 1: if classes > 1:
for i in range(classes): for i in range(classes):
ax[i + 1].set_title(f'Output mask (class {i + 1})') ax[i + 1].set_title(f'Output mask (class {i + 1})')
ax[i + 1].imshow(mask[:, :, i]) ax[i + 1].imshow(mask[1, :, :])
else: else:
ax[1].set_title(f'Output mask') ax[1].set_title(f'Output mask')
ax[1].imshow(mask) ax[1].imshow(mask)