Распечатать случайную выборку из загрузчика данных в PyTorch

У меня есть определенный набор данных, загруженный в загрузчик данных. Например, если я хочу сохранить 100 изображений из этого загрузчика данных, как мне перебрать загрузчик данных, чтобы сохранить их?


python pytorch dataset dataloader
person CaxiomX20    schedule 03.08.2020    source источник
comment
Привет @ CaxiomX20, добро пожаловать в SO! что ты уже испробовал? Пожалуйста, помогите нам, предоставив немного дополнительной информации   -  person M Z    schedule 03.08.2020

Ответы (1)


arrow_upward
0
arrow_downward

Я не совсем уверен, что вы пытаетесь сделать (возможно, отредактируйте свой вопрос), но, возможно, это поможет:

dataset = Dataset()
dataloader = torch.utils.data.DataLoader(
                 dataloader,
                 batch_size=32,
                 num_workers=1,
                 shuffle=True)

for samples, targets in dataloader:
    # 'sample' now is a batch of 32 (see batch-size above) elements of your dataset

Это то, что ты хотел? Надеюсь, что так :)

person Theodor Peifer    schedule 04.08.2020