首页 > 其他分享 >5、pytorch加载数据

5、pytorch加载数据

时间:2023-01-26 11:11:06浏览次数:29  
标签:迭代 标签 label pytorch 文件夹 加载 数据 图片

1、Dataset:

提供一种方式去获取数据及其label值

******实现的功能:

  (1)如何获取每一个数据及其label

  (2)告诉我们总共有多少的数据(因为神经网络经常会对一个数据迭代多次,只有知道总共有    多少数据才可以在后面的训练过程知道训练多少次,才能把整个数据集迭代完进行下一次的迭代。)

******图片的常用组织形式:

  (1)他的文件夹名称就是图片所对应的label

  (2)分两个文件夹存储,一个文件夹存放训练用的图片,而每个图片对应的标签放在另一个文件夹里用txt文件单独存储每个图片的坐标以及对应的标签label

  (3)直接把标签定义为图片的名称。

2、Dataloader:

送进网络的时候不是一个一个送进去的而往往都是有bachsize,几个几个送,进行打包压缩送进去的。

所以他是为了后面的网络提供不同的数据形式

标签:迭代,标签,label,pytorch,文件夹,加载,数据,图片
From: https://www.cnblogs.com/ar-boke/p/17067629.html

相关文章