diff --git a/src/utils/utils.py b/src/utils/utils.py deleted file mode 100644 index f1a0818..0000000 --- a/src/utils/utils.py +++ /dev/null @@ -1,17 +0,0 @@ -import matplotlib.pyplot as plt - - -def plot_img_and_mask(img, mask): - classes = mask.shape[0] if len(mask.shape) > 2 else 1 - fig, ax = plt.subplots(1, classes + 1) - ax[0].set_title("Input image") - ax[0].imshow(img) - if classes > 1: - for i in range(classes): - ax[i + 1].set_title(f"Output mask (class {i + 1})") - ax[i + 1].imshow(mask[1, :, :]) - else: - ax[1].set_title(f"Output mask") - ax[1].imshow(mask) - plt.xticks([]), plt.yticks([]) - plt.show()