You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
14 lines
365 B
14 lines
365 B
5 months ago
|
import matplotlib.pyplot as plt
|
||
|
|
|
||
|
|
|
||
|
|
def plot_img_and_mask(img, mask):
|
||
|
|
classes = mask.max() + 1
|
||
|
|
fig, ax = plt.subplots(1, classes + 1)
|
||
|
|
ax[0].set_title('Input image')
|
||
|
|
ax[0].imshow(img)
|
||
|
|
for i in range(classes):
|
||
|
|
ax[i + 1].set_title(f'Mask (class {i + 1})')
|
||
|
|
ax[i + 1].imshow(mask == i)
|
||
|
|
plt.xticks([]), plt.yticks([])
|
||
|
|
plt.show()
|