scatter_ 做 one-hot的一些要注意的点

1.

如当label_map的范围是[0,19]

 one-hot的通道大小为 torch.FloatTensor(bs,20,h,w)

一般label_map经过transfrom后,通过totensor操作,压缩到了[0,1]的范围,然后要再执行两步

label = label*255.0 (因为label不需要压缩到[0,1], 还是要恢复到[0,19]

label [label==255] = 20  这一步是排除一些图像中的不关系的像素,按道理来说label*255.0后不会得到255这个结果的,如果得到了就赋值为20,新的标签,意味着一个不相关的通道,但往往这一步是无效的,不会有==255的像素,也就不会被赋值为20

2.

今天把Label_map输入到transform再输出,发现label_map的最大值从19变成了20,后来发现是插值的问题,resize插值的时候用的Image.BICUBIC,然后插值之后的结果就改变了,换成Image.BILINEAR就可以了

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值