为什么我的自定义数据集在self.transform中失败了--对象没有属性 "transfrom

1 人关注

我试图创建一个自定义数据集,但在转换时失败了。

Below is my code:

transform = transforms.Compose([transforms.ToPILImage(),
                                transforms.Resize((255,255)),
                                #transforms.PILToTensor()])
                                transforms.ToTensor(),
                                transforms.Normalize(mean_img,std_img)]) 
    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'

这里出了什么问题,有什么办法可以解决这个问题吗?

1 个评论
You misspelled transform .
python
dataset
transform
torchvision
O.H
O.H
发布于 2021-01-01
1 个回答
Pierre.Vriens
Pierre.Vriens
发布于 2021-01-01
已采纳
0 人赞同