PyTorch学习-DataLoader

(1)体会DataLoader的使用

把数据加载到神经网络中,官方文档解读:torch.utils.data — PyTorch 2.0 documentation

PyTorch学习-DataLoader_第1张图片

 PyTorch学习-DataLoader_第2张图片

 随机抽取4张图片PyTorch学习-DataLoader_第3张图片

 把随机抽取出来的图片都进行img和target的打包PyTorch学习-DataLoader_第4张图片

 (2)tensorboard加入使用

PyTorch学习-DataLoader_第5张图片

PyTorch学习-DataLoader_第6张图片

 最后一步 不够64张,把drop_last设为True,就会舍去不够的

PyTorch学习-DataLoader_第7张图片

PyTorch学习-DataLoader_第8张图片

 (3)shuffle的使用

PyTorch学习-DataLoader_第9张图片

当shuffle = False时,取图片的顺序都一样;设置为True时,随机选取,图片顺序不一样。

PyTorch学习-DataLoader_第10张图片

 

你可能感兴趣的:(pytorch,学习,人工智能)