我正在尝试创建自定义数据集,但在转换中失败
下面是我的代码:
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'
这里出了什么问题,你知道如何解决这个问题吗
如this comment中所述:
因此,将
transfrom
更改为transform
。。。你的问题可能会解决相关问题 更多 >
编程相关推荐