qq_29423387的博客

import os from PIL import Image from torch.utils.data import Dataset # dataset有两个作用：1、加载每一个数据，并获取其label；2、用len（）查看数据集的长度 class MyData(Dataset): def __init__(self, root_dir, label_dir): # 初始化，为这个函数用来设置在类中的全局变量 self.root_dir = root_dir