在做语义分割的时候，最后在模型预测的时候得到的预测图是灰度图，且像素值是0到6（数据集的分割标签就是0到6），所以直接显示就是一张类似于全黑的图片，为了将不同的分割区域区分明显，可以将0到6的像素值分散到0到255之间，得到下面的图片：

但是这样的效果还是不太明显，所以将其变成彩图可能是一个更好的选择，使用matplotlib可以直接将灰度图显示为彩色索引图像，就是利用plt.imshow()函数，结果如下：

?