Torch_5(Dataset)

class torch.utils.data.Dataset	
def My_Dataset(Dataset):
	def __init__(self,data):
		self.data = data
	def __len__(self):
		return len(self.data)
	def __getitem__(self,idx):
		pass

自定义的数据集抽象类,至少包含以上内容,否则会报错。
_len_:提供了数据集大小;
_getitem_:支持整数索引。

torchvision.datasets.ImageFolder(root,[transform,target_transform])

一个用于处理图片的、通用的Dataset类。

其接受的数据以文件夹-png图片格式的文件的形式组织。
其展示如下:
在这里插入图片描述
该类包含了以下的成员变量:

  1. self.classes : 用list保存的类名;
['cat', 'dog']
  1. self.idx_to_class :类名对应的索引;
{'cat': 0, 'dog': 1}
  1. self.imgs :保存(img_path,class)的list。
[('./data/dogcat_2/cat/cat.12484.jpg', 0), 
('./data/dogcat_2/cat/cat.12485.jpg', 0), 
('./data/dogcat_2/cat/cat.12486.jpg', 0),
('./data/dogcat_2/cat/cat.12487.jpg', 0), 
('./data/dogcat_2/dog/dog.12496.jpg', 1), 
('./data/dogcat_2/dog/dog.12497.jpg', 1), 
('./data/dogcat_2/dog/dog.12498.jpg', 1), 
('./data/dogcat_2/dog/dog.12499.jpg', 1)]
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

右边是我女神

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值