Выгрузка данных изображения и загрузка с помощью загрузчика данных pytorch

Question

Я хочу сбросить данные, чтобы я мог загрузить их обратно для обучения моей модели.

Мой код вырезан для сброса данных:

for batch_idx, (image, label) in enumerate(dataloader):
    image, label = image.to(device), label.to(device)
    perturbed_image = attack.perturb(image, label)
    
    #---------- Classifier ----------
    predict_A = classifier(perturbed_image)
    pred_label = torch.max(predict_A.data, 1)[1]
    
    if pred_label != label:
        adv_data.append( (perturbed_image.to("cpu"), label.to("cpu")) )

Есть ли другой способ? может правильно сбросить его, чтобы загрузить в torch.utils.data.DataLoader.

Answer 1

Самый простой подход - использовать torch.save для сохранения фактических тензоров perturbed_image и label в виде двоичных файлов, а затем использовать custom Dataset. Обратите внимание, что сохраненные тензоры - это не отдельное изображение / этикетка, а скорее пакеты изображений / этикеток. Ваш новый обычай Dataset должен учитывать это.