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()这个函数的功能就是将上面两个地址进行拼接
标签:实战,__,os,label,dataset,root,dir From: https://www.cnblogs.com/ar-boke/p/17068221.html