qq_43398933的博客

转载 Rsesnet 残差网络

import torch import os,glob import random import csv from torch.utils.data import Dataset from PIL import Image from torchvision import transforms from torch.utils.data import DataLoader # 第一部分：通过三个步骤得到输出的tensor类型的数据 class Dataset_self(Dataset): .