Torchvision.datasets中的ImageFolder函数详解
前几天在看代码时遇到制作数据集的一条代码:train_datasets = datasets.ImageFolder(train_dir, transform = train_transforms)train_dataloader = torch.utils.data.DataLoader(train_datasets, batch_size = batch_size, shuffle = True)主要是对Torchvision.datasets中的ImageFolder函数的不理解通过查该函数的源
原创
2020-06-01 10:04:26 ·
7934 阅读 ·
1 评论