Merge pull request #348 from pleaseRedo/master

Update utils.py. minor fix Former-commit-id: 1c736851c43aceb70f49f6e444f1e22a377ec59d
2022-03-31 21:37:48 +02:00 · 2022-03-31 21:37:48 +02:00 · c1c7c952aa
parent 2ca43802cc eb95dab01e
commit c1c7c952aa
1 changed files with 1 additions and 1 deletions
--- a/utils/utils.py
+++ b/utils/utils.py
@ -9,7 +9,7 @@ def plot_img_and_mask(img, mask):
    if classes > 1:
        for i in range(classes):
            ax[i + 1].set_title(f'Output mask (class {i + 1})')
-            ax[i + 1].imshow(mask[:, :, i])
+            ax[i + 1].imshow(mask[1, :, :])
    else:
        ax[1].set_title(f'Output mask')
        ax[1].imshow(mask)