Python输出的标签图处理：单通道的tensor转化为三通道的彩色图

跑代码的时候被一个图像转换绊了两天，记录一下。

起因为跑的一个代码上面输出的结果是全黑的标签图，也没有设置保存。

刚开始的尝试是先输出灰度图，发现输出的图像不太理想，看不出啥来，按照很多教程看了看都没跑通。

需要转换的predict图

这里要处理的单通道的图像，输出的label为0，1，2，3这样的，整体的图像是黑色，为tensor,type为int64，并且输出是一个batch的图

图中是看不出颜色变化的，打印出来是这样的

显示为单通道模式的tensor，这里的[4,512,512]指的是4个图拼一起（因为输出是一个batch，即4张图），512X512可以理解为512个0 1 2 3组起来。（这里的数据是自定义的语义分割数据集，加上背景4个classes）

解决办法：

解决如下：

先定义一个palette，这里是像素点为0 1 2 3所对应的三通道像素

 都写在注释里了

这样一个三通道的丑图就诞生了