我正在处理语义分割任务，我的目标以及网络输出都是0到10的整数标签矩阵（在这种情况下为84x64）。我想将这些矩阵打印为图像，每种颜色标签。我该怎么办？
我正在使用matplotlib的imshow函数。我尝试为该图设置不同的cmap（顺序图和定性图），但是由于每个矩阵中只有3-4个不同的值，因此同一标签的颜色因图像而异，如{{3}中所示}。
手动检查矩阵时，标签正确，这意味着它们等于显示的数字。
我考虑过将每个标签转换为RGB值以得到84x64x3的图像，但是我觉得必须使用一个更简单的方法。