plt.imshow 全黑解决办法

# 标签有时候数据偏小，需要给赋予其他颜色方便可视化
image_file = "/root/autodl-tmp/datasets/consep/labels/train/000.png"

classes = ('background','Epithelial',
'Inflammatory',
'Spindle-Shaped',
'Miscellaneous')
palette = [[0, 0, 0], [129, 127, 38], [120, 69, 125], [53, 125, 34], 
           [0, 11, 123]]
image = Image.open(image_file).convert('P')
image.putpalette(np.array(palette, dtype=np.uint8))
image

这个问题根本原因是数据偏小，比如只有6类，则最大取值为5，而0是全黑，255 是白色，就导致整张图看上去都是黑色。
这个方案是赋予其他颜色进行可视化。
如有其他方案，后续补充