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

語義分割做法: 之前一直疑惑語義分割是怎麼做到顏色這麼分明的,後來看了代碼。 我這邊這個語義分割是8類的分割,所以圖片經過語義分割後輸出的網絡是8*H*W的,裏面的8就代表8類,然後用torch.max選出每個位置上值最大的索引,然後根據索引對那個位置的像素圖上對應的顏色。
相關文章
相關標籤/搜索