pytorch数据dataset的三种读取方式

文章目录

    • 1.自带的datasets
    • 2.ImageFolder
    • 3.自己创建的dataset(用的多)

1.自带的datasets

  • pytorch自带的数据集
    具体数据集如下:
    pytorch数据dataset的三种读取方式_第1张图片
    用法:
    pytorch数据dataset的三种读取方式_第2张图片

  • 可以用的功能
    pytorch数据dataset的三种读取方式_第3张图片
    pytorch数据dataset的三种读取方式_第4张图片

2.ImageFolder

文件形式如下时用此:(一个文件下只有一个类别)
pytorch数据dataset的三种读取方式_第5张图片
例如:天气数据集data 下有 train 跟test
train下有四个类别(多云、晴、雨、雪) test下有四个类别 例如 多云文件下 全是多云的图片
pytorch数据dataset的三种读取方式_第6张图片

后面的用法跟pytorch自带数据集的用法一样。

3.自己创建的dataset(用的多)

  • 可以根据的需求,随意添加

pytorch数据dataset的三种读取方式_第7张图片
pytorch数据dataset的三种读取方式_第8张图片
pytorch数据dataset的三种读取方式_第9张图片
pytorch数据dataset的三种读取方式_第10张图片

  • 代码如下:
    pytorch数据dataset的三种读取方式_第11张图片
    pytorch数据dataset的三种读取方式_第12张图片
    pytorch数据dataset的三种读取方式_第13张图片
    pytorch数据dataset的三种读取方式_第14张图片
    pytorch数据dataset的三种读取方式_第15张图片

你可能感兴趣的:(pytorch,pytorch,深度学习,python)