人工智能学习06--pytorch05--torchvision中的数据集使用&DataLoader的使用

torchvision中的数据集使用

在这里插入图片描述
test_set的class属性
在这里插入图片描述
在这里插入图片描述
把数据集每一部分都变成tensor类型
现在输出的就是tensor数据类型了
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

DataLoader的使用

在这里插入图片描述

在这里插入图片描述

  • batch_size 一摞牌中,每次抓几张
  • shuffle 打乱,第二次打牌前,牌的顺序要跟第一次不一样(默认False:跟之前排序一样;一般设置为True)
  • num_works 多进程加载,默认0 (电脑核数不够,num_workers得调低)num_worker=0是指只有一个主进程
    在这里插入图片描述
  • drop_last 若一共一百张牌,每次取三张,最后会剩下一张,是否舍去它(True:只要前面99张牌)

在这里插入图片描述
在这里插入图片描述
target不是标签,是标签存放的位置,标签列表是calsses,真正的标签是classes[target]
在这里插入图片描述
第几次取数据的时候,数据长什么样
在这里插入图片描述
在这里插入图片描述
剩余的数据集只有16张,不足64张
在这里插入图片描述
将drop_last设置为True,可以看到把数据集最后的部分进行舍去
(一定要给add.images里面改名!)
在这里插入图片描述

把shuffle改为False
可以看到取的都是一样的
在这里插入图片描述

把shuffle改为True
在这里插入图片描述
在这里插入图片描述
一般把imgs作为神经网络的输入

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值