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