how to apply different transforms to pytorch dataset subsets

GHD answered on January 4, 2023 Popularity 1/10 Helpfulness 1/10

answer how to apply different transforms to pytorch dataset subsets

how to apply different transforms to pytorch dataset subsets

Comment

from copy import copy

train_dataset, test_dataset = random_split(full_dataset, [train_size, test_size])
train_dataset.dataset = copy(full_dataset)

test_dataset.dataset.transform = transforms.Compose([
    transforms.Resize(img_resolution),
    transforms.ToTensor(),
    transforms.Normalize(mean=[0.485, 0.456, 0.406],
                         std=[0.229, 0.224, 0.225])
])

train_dataset.dataset.transform = transforms.Compose([
    transforms.RandomResizedCrop(img_resolution[0]),
    transforms.ToTensor(),
    transforms.Normalize(mean=[0.485, 0.456, 0.406],
                         std=[0.229, 0.224, 0.225])
])

Popularity 1/10 Helpfulness 1/10 Language python

Source: stackoverflow.com

Tags: apply dataset python pytorch

Link to this answer
Share Copy Link