天津网站建设排名百度开户返点

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_