神经网络入门实战：（五）本地数据集的读取，以及从pytorch官网下载数据集的操作

本地带标签图片数据集的读取

代码里面的注释写的都比较清楚，也有拓展的部分，这里就不详细列出。

from torch.utils.data import Dataset # 从torch.utils.data导入Dataset类，用于创建自定义的数据集类
from PIL import Image #引入PIL库中的Image模块，进行图片的读取、处理
import os # 引入os模块，用于文件和目录路径的操作

class MyDataset(Dataset):

	def __init__(self,root_dir,label_dir):
		self.root_dir = root_dir # 图片路径的前一部分
		self.label_dir = label_dir # 图片的标签，同时也是路径中最后一层文件夹的名称
		self.path = os.path.join(self.root_dir,self.label_dir) # 合并上述两个部分，组成完整文件夹路径名<