Как сделать 3D-тензор в кодировке one_hot

Question

Например,

, метки - это истинная правда, метки размер - [4,224,224], где 4 означает размер пакета, 224 - h и w, метки dtype - это torch.int64, в моем обучающем коде пиксель метки равен 0 до 255. Я использую свою сеть для Semanti c сегментация

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

Надеюсь, это поможет!