mirror of
https://github.com/Laurent2916/REVA-QCAV.git
synced 2024-11-08 14:39:00 +00:00
chore: delete unused file
Former-commit-id: 540acfba838755064facc7a39bfe72b6cc0a2796 [formerly 085dd013a8e439d44b173a563b0fa8cd40cd9b75] Former-commit-id: c6c8fffc2768ec54ba1adb1648de17415f130db7
This commit is contained in:
parent
806e98c744
commit
9b624da832
|
@ -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()
|
Loading…
Reference in a new issue