1

我想在张量的第四层中向图像添加更多信息,前三层基于图像。之后,我想从图像中删除和平(数据增强),并且必须将图像调整为给定大小。

为此,我从一张图片中创建了一个张量,并使用 torch.cat 将它与一个带有一层附加信息的张量连接起来。(几乎但不是所有第二张量的条目都是零。)

我通过 transforms.compose 发送了结果(以剪切和调整张量的大小)但之后张量完全由零组成。

在这里,我建立了一个可重现的示例。

import torch
from torchvision import transforms

height = 2
width = 4
resize = 2
tensor3 = torch.rand(3,height,width)
tensor1 = torch.zeros(1,height,width)
#tensor1 = torch.rand(1,height,width)

imageToTensor = transforms.ToTensor()
tensorToImage = transforms.ToPILImage()

train_transform = transforms.Compose([
    transforms.RandomResizedCrop(resize, scale=(0.9, 1.0)),
    transforms.ToTensor(),
])

tensor4 = torch.cat((tensor3,tensor1),0)

image4 = tensorToImage(tensor4)
transformed_image4 = train_transform(image4)

print(tensor4)
print(transformed_image4)
tensor([[[0.6774, 0.5293, 0.4420, 0.2463],
         [0.1391, 0.7481, 0.3436, 0.9391]],

        [[0.0652, 0.2061, 0.2931, 0.6126],
         [0.2618, 0.3506, 0.5095, 0.7351]],

        [[0.8555, 0.6320, 0.9461, 0.0928],
         [0.2094, 0.3944, 0.0528, 0.7900]],

        [[0.0000, 0.0000, 0.0000, 0.0000],
         [0.0000, 0.0000, 0.0000, 0.0000]]])

tensor([[[0., 0.],
         [0., 0.]],

        [[0., 0.],
         [0., 0.]],

        [[0., 0.],
         [0., 0.]],

        [[0., 0.],
         [0., 0.]]])

如果我选择“tensor1 = torch.rand(1,height,width)”,我就没有这个问题。但如果大多数条目都是零,我有。使用 scale=(0.5, 1.0) 我也没有问题。

没有一些问题:

  1. 如何使用非零条目调整前三层的大小?

  2. 是我误会了什么,还是真的很奇怪?

4

1 回答 1

0

我创建了一个问题:

https://github.com/pytorch/pytorch/issues/22611

答案是 Torchvision 只支持 PIL-Images。

另一种方法是用于转换的albumentations-library。

于 2019-07-13T14:08:51.570 回答