【数据】pytorch的官方自带数据

本站原创文章，转载请说明来自《老饼讲解-深度学习》www.bbbdata.com

本文介绍pytorch提供了哪些数据，在哪里下载，和下载后得到的数据格式

并展示一个具体的示例，说明数据的下载流程和下载后得到什么样的数据

一、pytorch的数据介绍   

本节简单介绍pytorch提供了哪些数据，在哪里下载

01. pytorch提供了哪些数据

pytorch的torchvision.datasets提供了许多自带图片、文本、视频数据集，

具体链接地址如下：

👉 图 片 数 据： https://pytorch.org/vision/stable/datasets.html 

👉 文 本 数 据： https://pytorch.org/text/stable/datasets.html 

👉 视 频 数 据： https://pytorch.org/audio/stable/datasets.html 

02. pytorch提供的数据是什么格式

pytorch提供的数据集是pytorch中的DataSet的格式，

下载后就可以装载到DataLoader中进行使用

二、怎么下载pytorch的数据   

本节具体展示一个示例，怎么下载pytorch的数据，和下载后的数据长什么样子

01.如何下载pytorch的自带数据

以图片数据为例，先进入pytorch的图片数据列表页面 https://pytorch.org/vision/stable/datasets.html ，

然后找到需要加载的数据集，点击进入具体的数据集页面，查看相关的API说明，根据说明配置来编写代码进行下载

例如需要下载Flowers102数据，就点击Flowers102进去查看相关的API说明,如下：

根据API说明配置好数据获取的接口，运行后就可以进行下载，

示例如下：

from torchvision import datasets
from torchvision.transforms import ToTensor

root= 'D:\pytorch\data'            # 路径,用于存放下载的数据(需要改成自己的目录)
img_data = datasets.Flowers102(
    root       = root              # 路径,如果路径有,就直接从路径中加载,如果没有,就联网获取
    ,split     ='train'
    ,transform = ToTensor()        # 转换为tensor数据
    ,download  = True              # 是否下载,选为True,就下载到root下面
    ,target_transform= None)

每个数据的下载接口会略为不同，下张文章再具体讲解每个参数的意义，和更完整的下载示例

02.下载后得到什么样的数据

运行上述代码后，得到的img_data变量如下：

返回的img_data变量可以直接装载到DataLoader中进行使用

同时，硬盘中下载到的数据如下：

 End