如何使用pytorch对我的数据集应用2种不同的转换

s2j5cfk0 于 2023-03-23 发布在其他

关注(0)|答案(1)|浏览(88)

我正在为pytorch中的下游CNN应用程序准备数据。我希望在同一个数据集上使用transform方法= ToTensor（）和Resize（）。我无法将它们一起应用。

from torchvision.datasets import ImageFolder
from torchvision.transforms import ToTensor # In order to convert the datat to pytorch tensors
from torchvision.transforms import Resize

可以单独应用它们，但需要在同一数据集上应用这两种转换。图像文件存储在"/content/sub_folder"目录中。

dataset = ImageFolder("/content/sub_folder" , transform = Resize([224,224])) # ToTensor is method

dataset = ImageFolder("/content/sub_folder" , transform = ToTensor()) # ToTensor is method

如何在我的文件夹中包含的所有图像上应用这两种转换。

pytorch

来源：https://stackoverflow.com/questions/75807711/how-to-apply-2-different-transformation-on-my-dataset-using-pytorch

1条答案

按热度按时间

eqqqjvef1#

你可以使用compose来使用多个转换

from torchvision import transforms as T
from torchvision.datasets import ImageFolder

trans_comp = T.Compose([
  T.Resize([224,224]),
  T.ToTensor()
])

dataset = ImageFolder("/content/sub_folder", transform=trans_comp)

赞(0）回复(0）举报 2023-03-23

我来回答

如何使用pytorch对我的数据集应用2种不同的转换

1条答案

相关问题

热门标签

最新问答