5、pytorch加载数据
1、Dataset:
提供一种方式去获取数据及其label值
******实现的功能:
(1)如何获取每一个数据及其label
(2)告诉我们总共有多少的数据(因为神经网络经常会对一个数据迭代多次,只有知道总共有 多少数据才可以在后面的训练过程知道训练多少次,才能把整个数据集迭代完进行下一次的迭代。)
******图片的常用组织形式:
(1)他的文件夹名称就是图片所对应的label
(2)分两个文件夹存储,一个文件夹存放训练用的图片,而每个图片对应的标签放在另一个文件夹里用txt文件单独存储每个图片的坐标以及对应的标签label
(3)直接把标签定义为图片的名称。
2、Dataloader:
送进网络的时候不是一个一个送进去的而往往都是有bachsize,几个几个送,进行打包压缩送进去的。
所以他是为了后面的网络提供不同的数据形式

浙公网安备 33010602011771号