pytorch数据dataset的三种读取方式

1.自带的datasets

  • pytorch自带的数据集
    具体数据集如下:
    在这里插入图片描述
    用法:
    在这里插入图片描述

  • 可以用的功能
    在这里插入图片描述
    在这里插入图片描述

2.ImageFolder

文件形式如下时用此:(一个文件下只有一个类别)
在这里插入图片描述
例如:天气数据集data 下有 train 跟test
train下有四个类别(多云、晴、雨、雪) test下有四个类别 例如 多云文件下 全是多云的图片
在这里插入图片描述

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

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

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

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

  • 代码如下:
    在这里插入图片描述
    在这里插入图片描述
    在这里插入图片描述
    在这里插入图片描述
    在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/weixin_41504611/article/details/128553191