Загрузчик данных Pytorch Преобразует ошибку тензора

[Класс набора данных][1]Преобразования, примененные к изображению Невозможно получить доступ к значениям загрузчика данных pytorch для использования TypeError: default_collate: пакет должен содержать тензоры, массивы numpy, числа, словари или списки; найденный объект

Ошибку можно воспроизвести, запустив https://jovian.ml/pravin-bnmit/cxr/ на kaggle введите здесь описание изображения


python pytorch dataloader image-preprocessing
person Nemaly Praveen    schedule 15.07.2020    source источник
comment
Я думаю, было бы полезно, если бы вы поделились большим количеством кода. действительно трудно предположить, в чем проблема. Я предполагаю, что изображение не преобразуется в тензор (информация: pytorch.org/docs /stable/torchvision/transforms.html)   -  person Victor Zuanazzi    schedule 15.07.2020
comment
Я добавил код, и блокнот доступен по адресу jovian.ml/pravin-bnmit/cxr   -  person Nemaly Praveen    schedule 15.07.2020
comment
ошибки в блокноте и на изображении, которое вы здесь разместили, разные. ToTensor решил вашу проблему? (загрузчики данных не подлежат подписке, вы можете сделать next(dl) для загрузки следующей партии или просто for data in dl: do something with data)   -  person Victor Zuanazzi    schedule 15.07.2020
comment
@VictorZuanazzi Ошибка связана с циклом for, когда я пытаюсь получить доступ к данным из dl. Это дает ошибку, показанную выше. Вы можете проверить блокнот, который я обновил.   -  person Nemaly Praveen    schedule 15.07.2020

Ответы (1)


arrow_upward
0
arrow_downward

Почему бы вам не попробовать img = img.resize((1024, 1024)) перед преобразованием внутри метода getitem()?

def __getitem__(self, idx):
    row = self.df.loc[idx]
    img_id, img_label = row['Image Index'], row['disease_vec']
    img_fname = row['path']
    img = Image.open(img_fname)
    img = img.resize((1024, 1024))
    if self.transform:
        img = self.transform(img)
    return img, img_label
person planet_pluto    schedule 15.07.2020
comment
Проблема в том, что он возвращает кортеж при выходе из класса набора данных. который является объектом, и цикл for хочет получить доступ к 2 тензорам, которые он не может вызвать, вызывая ошибку - person Nemaly Praveen; 16.07.2020