datasetdataloader专题

手撸AI-1:构建DatasetDataloader,搭建模型训练基础架构

一. 构建Dataset 构建Dataset无非就是创建一个类继承dataset并重写三个方法: from torch.utils.data import Datasetfrom PIL import Imageclass MyDataset(Dataset):def __init__(self):# 初始化数据集passdef __getitem__(self, index):# 根据索引