文章目录 utils_dataset.pymain-调用utils_dateset.py utils_dataset.py ''' 1默认:没有改变尺寸,数据集中的图像可以是任意形状尺寸。dataloader中必须令batch_size=1 transforms.Resize((宽,高))(image) 和 batch_size=1 必须用其一 原因:当batch_size>1时,每个batch的数据必须一样大,需要transforms.Resize来统一图像尺寸。如果图片数据不一样大,又不想resize改变图像尺寸,就令batch_size=12··· 3··· '''import torch import numpy as np import matplotlib.pyplot as plt import glob import PIL from torchvision import transforms from torchvision.datasets import ImageFolder from torch.utils.data import Dataset, DataLoaderclass MyDataset(Dataset):def __init__(self, img_path1, img_path2): # 输入参数:输入图像的路径list,label图像的路径listself.path1 = img_