为什么自定义数据集在self中失败。transform对象没有属性“transfrom”

2024-09-20 22:25:01 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在尝试创建自定义数据集,但在转换中失败

下面是我的代码:

transform = transforms.Compose([transforms.ToPILImage(),
                                transforms.Resize((255,255)),
                                #transforms.PILToTensor()])
                                transforms.ToTensor(),
                                transforms.Normalize(mean_img,std_img)]) 





class img_dataset_fun(Dataset):

    def __init__(self, csv_file, transform):
        self.csv_file=pd.read_csv(csv_file)
        self.transform = transform

    def __len__(self):
        return len(self.csv_file)

    def __getitem__(self, index):
        if torch.is_tensor(index):
            index = index.tolist()
        img_path=self.csv_file.iloc[index,1]
        image=io.imread(img_path)
        if self.transform is not None:
          image=self.transfrom(image)

      
        return image

img_dataset = img_dataset_fun(csv_file='data.csv',transform= transform)

sample = img_dataset[3]

尝试运行示例时,出现以下错误:

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-91-5dd830a096ff> in <module>()
----> 1 sample = img_dataset[3]

<ipython-input-88-682960274e67> in __getitem__(self, index)
     17         if self.transform is not None:
     
---> 18           image=self.transfrom(image)
   

AttributeError: 'img_dataset_fun' object has no attribute 'transfrom'

这里出了什么问题,你知道如何解决这个问题吗


Tags: csvimageselfimgindexlenifis

热门问题