如何將 3D 張量制作成 one_hot 編碼

Question

例如，labels 是一個groundtruth， labels size 是[4,224,224]，其中4 表示batchsize，224 表示h 和w，labels dtype 是torch.int64，在我的訓練代碼中，label 的像素從0 到255。我使用我的網絡來語義化分割

Answer 1

從您的問題來看，您似乎想要 256 個整數標簽，從 0 到 255。這可以通過以下方式完成：

# let a be the (4, 224, 224) dim tensor
labels = np.stack([(a == i).int() for i in range(256)])
print(labels.shape) #(256, 4, 224, 224) dimensional one-hot encoding

希望這可以幫助！