如何解决为什么我的自定义数据集在 self.transform 中失败 - 对象没有属性 'transfrom'
我正在尝试创建自定义数据集,但转换失败。
下面是我的代码:
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'
这里有什么问题,知道如何解决这个问题吗?
解决方法
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。