Dataset类的使用
from torch.utils.dataimportDatasetclassMyData(Dataset):def__init__(self,root_dir,label_dir):...def__getitem__(self,idx):...from torch.utils.dataimportDatasetclassMyData(Dataset):def__init__(self,root_dir,label_dir):...def__getitem__(self,idx):...