文章目录
- 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=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_
