首页
Python
Java
前端
数据库
Linux
Chatgpt专题
开发者工具箱
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):# 根据索引
阅读更多...