9-2-Dataset创建-import调用

文章目录 utils_dataset.py main-调用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=1 2··· 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, DataLoader class MyDataset(Dataset): def __init__(self, img_path1, img_path2): # 输入参数：输入图像的路径list，label图像的路径list self.path1 = img_