ff1ac0936c118d129bc8a8014958948d3b3883be,utils/data_vis.py,,plot_img_and_mask,#Any#Any#,3

Before Change

    a.set_title("Input image")
    plt.imshow(img)

    b = fig.add_subplot(1, 2, 2)
    b.set_title("Output mask")
    plt.imshow(mask)
    plt.show()

After Change

def plot_img_and_mask(img, mask):
    classes = mask.shape[2] 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:

Instances