分类：自定义训练数据

2021
03-05

pytorch Dataset,DataLoader产生自定义的训练数据案例

1.torch.utils.data.Datasetdatasets这是一个pytorch定义的dataset的源码集合。下面是一个自定义Datasets的基本框架，初始化放在__init__()中，其中__getitem__()和__len__()两个方法是必须重写的。__getitem__()返回训练数据，如图片和label，而__len__()返回数据长度。classCustomDataset(data.Dataset):#需要继承data.Datasetdef__init__(self):#TODO#1.Initializefilepathorlistoffilenames.passdef__ge... 继续阅读 >