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

语义分割做法: 之前一直疑惑语义分割是怎么做到颜色这么分明的,后来看了代码。 我这边这个语义分割是8类的分割,所以图片经过语义分割后输出的网络是8*H*W的,里面的8就代表8类,然后用torch.max选出每个位置上值最大的索引,然后根据索引对那个位置的像素图上对应的颜色。
相关文章
相关标签/搜索