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作为神经网络的输入