weixin_44003314的博客

【读取文件】 【路径】 from torch.utils.data import Dataset from PIL import Image import os class MyData(Dataset): # 继承Dataset def __init__(self, root_dir,label_dir): self.root_dir = root_dir # self.label_dir = label_dir self.path =