自定义数据集 一、使用torchvision.io读取照片 二、使用PIL读取照片 一、使用torchvision.io读取照片 import numpy as np import torch from PIL import Image import numpy from matplotlib import pyplot as plt from torch.utils.data import Dataset,DataLoader from torchvision import transforms,utils import warnings