Dataset&DataLoader
dataset需要实现def __getitem__(self, index) -> T_co: # 基于一个索引返回一个训练样本(x, y)构成的训练对
raise NotImplementedError("Subclasses of Dataset should implement __getitem__.")
def __len__(self): # 将数据放进去,返回数据总大小 return len(self.img_labels)
return self.tensors[0].size(0)import os
import pandas as pd
from torchvision.io import read_image
# 从磁盘中读取训练数据、__getitem__中能够根据idx返回相对...