Pytorch教程

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

作者 : 老饼 发表日期 : 2023-07-28 10:47:37 更新日期 : 2023-12-04 04:40:38
本站原创文章,转载请说明来自《老饼讲解-深度学习》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 







联系老饼