Преобразования PyTorch изменяют размеры

Question

Я не понимаю, почему преобразования pytorch из 100x100 рис. Делают 3x100 рис.

print("Original shape ", x.shape)

x = transforms.Compose([
    transforms.ToPILImage(),
    transforms.ToTensor()
])(x)

print("After transformation shape ", x.shape)

выходы

Original shape  torch.Size([100, 100, 3])
After transformation shape  torch.Size([3, 100, 3])

Что происходит?

Answer 1

Согласно документам https://pytorch.org/docs/stable/torchvision/transforms.html#torchvision.transforms.ToPILImage,, если входной сигнал является тензором Факела, форма имеет вид C x H x W. Таким образом, 100 считается числом каналов. Поскольку нет режима, который соответствует 100 каналам, он интерпретируется как RGB (3 канала).

Итак, вам нужна форма ввода torch.Size([3, 100, 100]), чтобы она работала так, как вы хотите.