語義分割是如何做到顏色分明的

語義分割做法：

之前一直疑惑語義分割是怎麼做到顏色這麼分明的，後來看了代碼。

我這邊這個語義分割是8類的分割，所以圖片經過語義分割後輸出的網絡是8*H*W的，裏面的8就代表8類，然後用torch.max選出每個位置上值最大的索引，然後根據索引對那個位置的像素圖上對應的顏色。