6、dataset实战

1、首先从包中导入dataset类（from torch.utils.data import Dataset）

2、定义一个子类去继承dataset类，可以重写里面的方法

　　（1）__init__(self):初始化方法（必写，因为创建一个实例的时候必须要进行初始化）

　　（2）__getitem__(self，item):获取每一条数据的方法

　　　　可以在pycharm下面的python console控制台里来边编写边查看

  

首先获取图片地址（通过索引来获取，必须定义个列表来存放这些地址）----------import os  引入os库，

 

 

 

def __init__(self,root_dir,label_dir):

# 为整个类提供全局变量，root_dir用来获取存放图片的文件夹地址，label_dir代表图片标签的文件夹地址

root_dir="learn_Pytorch/dataset/hymenoptera_data/train"
label_dir="learn_Pytorch/dataset/hymenoptera_data/train/ants"
path=os.path.join(root_dir,label_dir)；    os.path.join（）这个函数的功能就是将上面两个地址进行拼接