from torchvision import transforms, utils
from torchvision import datasets
import torch
import matplotlib.pyplot as plt
train_data = datasets.ImageFolder('./images_train', transform=transforms.Compose([
transforms.RandomResizedCrop(224),
transforms.RandomHorizontalFlip(),
transforms.ToTensor()
]))
train_loader = torch.utils.data.DataLoader(train_data,
batch_size=8,
shuffle=True
用ImageFolder上传子文件图片数据集
最新推荐文章于 2024-07-16 18:10:07 发布
本文详细介绍了如何利用Python的torchvision库中的ImageFolder模块,有效地加载和处理包含多个子目录的图像数据集。每个子目录代表一个类别,ImageFolder会自动根据目录结构进行分类。通过实例代码,展示了如何预处理图像、调整尺寸以及应用数据增强技术,以优化深度学习模型的训练过程。
摘要由CSDN通过智能技术生成