Pytorch(4)-DataLoader的使用
2022/5/22 23:06:50
本文主要是介绍Pytorch(4)-DataLoader的使用,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!
# DataLoader:数据加载器,从dataset中取数据,具体怎么取,取什么都通过dataLoader来完成,dataLoader中除了dataset参数外其他参数都有默认值 # dataset:从哪里加载数据 # batch_size:一次取出多少数据,如果为2,则一次从数据集中取出两条数据 # shuffle:数据集是不是打乱的意思,就比如打牌,第一局打完了,第二局的牌的顺序是否和第一局一样,如果为true代表两局牌的顺序不一样,否则一样,一般设置为true # num_workers:是否采用多个进程加载程序 # drop_last:如果除不尽时,数据集中最后一个数据是否要舍弃,如果为true则舍弃,否则不舍弃 import torchvision from torch.utils.data import DataLoader from torch.utils.tensorboard import SummaryWriter # 准备测试的数据集 test_data = torchvision.datasets.CIFAR10(root="F:\\pytorch\\pytorch01_hello\\dataset\\train\\torchvision_image", train=False, transform=torchvision.transforms.ToTensor()) # 测试DataLoader test_dataLoader = DataLoader(dataset=test_data, batch_size=4, shuffle=True, num_workers=0, drop_last=False) # 查看数据集中第一张图片大小及其target img, target = test_data[0] print(img.shape) print(target) print("================================") # 查看数据集中多张图片的大小及其target writer = SummaryWriter("dataloader") step = 0 for data in test_dataLoader: imgs, targets = data writer.add_images("test_dataLoader", imgs, step) step = step+1 print(imgs.shape) print(targets) writer.close()
这篇关于Pytorch(4)-DataLoader的使用的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!
- 2024-07-05feign默认connecttimeout和readtimeout是多少-icode9专业技术文章分享
- 2024-07-05idea控制台,日志太多,导致部分想看得日志被刷走 搜不到-icode9专业技术文章分享
- 2024-07-05The server selected protocol version Tls10 is not accepted by client preferences [TLs12]-icode9专业技术文章分享
- 2024-07-05怎么清理项目缓存-icode9专业技术文章分享
- 2024-07-04安装 Eyoucms详细图文教程-icode9专业技术文章分享
- 2024-07-04ueditor 复制文章时,图片的链接是一个下载图片地址,该如何处理?-icode9专业技术文章分享
- 2024-07-04怎样判断host有没有对wordpress有缓存呢-icode9专业技术文章分享
- 2024-07-04具有编译功能的系统make后,无法ssh连接-icode9专业技术文章分享
- 2024-07-04make后如何升级ssh-icode9专业技术文章分享
- 2024-07-03微信支付提示下单账户与支付账户不一致-icode9专业技术文章分享