show_images.py 文件源码

def vis_col_im(im, gt):
    indices_0 = np.where(gt == 0) # nothing
    indices_1 = np.where(gt == 1) # necrosis
    indices_2 = np.where(gt == 2) # edema
    indices_3 = np.where(gt == 3) # non-enhancing tumor
    indices_4 = np.where(gt == 4) # enhancing tumor

    im = np.asarray(im, dtype='float32')
    im = im*1./im.max()
    rgb_image = color.gray2rgb(im)
    m0 = [1., 1., 1.]
    m1 = [1., 0., 0.]
    m2 = [0.2, 1., 0.2]
    m3 = [1., 1., 0.2]
    m4 = [1., 0.6, 0.2]

    im = rgb_image.copy()
    im[indices_0[0], indices_0[1], :] *= m0
    im[indices_1[0], indices_1[1], :] *= m1
    im[indices_2[0], indices_2[1], :] *= m2
    im[indices_3[0], indices_3[1], :] *= m3
    im[indices_4[0], indices_4[1], :] *= m4

    plt.imshow(im)
    plt.show()
    plt.close()