图像处理数据集预处理

RockLiu@805

已于 2025-03-18 18:13:05 修改

阅读量636

点赞数 17

分类专栏：机器视觉文章标签：深度学习机器学习人工智能

于 2024-04-10 21:40:06 首次发布

本文链接：https://blog.csdn.net/qq_18943707/article/details/137611848

版权

多分类数据集的dataset操作

对于文件夹名字是训练模型所需label时，如下图

我们可以通过使用torchvision.datasets.ImageFolder来对我们的多个dataset进行数据加载

train_transforms_dataset = torchvision.datasets.ImageFolder(root=train_path, transform=transforms_img)

加载完成后即可使用DataLoader进行数据分批操作

train_transforms_loader = DataLoader(train_tran