pytorch中几种Dataset读取数据重写实现
通过重写Dataset类,对自己制作的数据集进行读取后传给DataLoader。主要用来完成从哪里读取数据和标签的功能。主要是__getitem__(返回数据集和标签)和__len__(返回数据的长度)这两个方法。import numpy as npimport torchimport osfrom PIL import Imagefrom torch.utils.data import Datasetclass MyDataset_1(Dataset): """ 通过包含数据
原创
2021-06-07 15:36:23 ·
2043 阅读 ·
0 评论