语义分割是如何做到颜色分明的

语义分割做法：

之前一直疑惑语义分割是怎么做到颜色这么分明的，后来看了代码。

我这边这个语义分割是8类的分割，所以图片经过语义分割后输出的网络是8*H*W的，里面的8就代表8类，然后用torch.max选出每个位置上值最大的索引，然后根据索引对那个位置的像素图上对应的颜色。