Pytorch学习-dataset

参考视频:【PyTorch深度学习快速入门教程(绝对通俗易懂!)【小土堆】】 https://www.bilibili.com/video/BV1hE411t7RN/?share_source=copy_web&vd_source=94395bc3053afb19a3884017d26e997f

一、dataset类的解释

运行jupyter

首先,输入:

from torch.utils.data import Dataset

意思是:从torch这个“大工具箱里”的“常用工具区”中的关于数据的data区import Dataset

(jupyter中使用“shift+回车”运行当前块)

其次,使用以下两种方法查看“Dataset”的官方文档如何解释

 二、代码实战

数据集图片 在控制台的查看和使用

将数据集放到项目里面

复制图片路径查看:

 获取图片列表

 创建“art”数据集

再创建一个“house”数据集

 取第一个数据

 合并数据集

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
Pytorch中遍历dataset可以使用torch.utils.data.DataLoader这个类。在初始化DataLoader时,一般常用的参数有dataset、batch_size、shuffle和num_workers等。其中dataset就是我们构建的自定义dataset类。在使用时,可以直接使用for循环来遍历dataloader对象,并且可以通过迭代器的方式输出每个batch的数据。具体实现如下: ```python import torch from torch.utils.data import DataLoader # 创建自定义的dataset对象 dataset = MyDataset() # 创建dataloader对象,并指定batch_size和是否进行数据打乱 dataloader = DataLoader(dataset, batch_size=32, shuffle=True) # 遍历dataloader对象 for batch_data in dataloader: # 处理每个batch的数据 inputs, labels = batch_data # 进行模型的训练或预测等操作 ... ``` 在遍历dataloader时,实际上是从dataset中取出数据,只是在取数据的规则上进行了一些修改,比如可以进行数据的打乱操作。因此,在遍历dataloader时,会调用自己定义的dataset类中的__getitem__()方法来获取数据。通过这种方式,我们可以方便地对数据进行mini-batch的训练。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* *3* [使用Pytorch中的Dataset类构建数据集的方法及其底层逻辑](https://blog.csdn.net/rowevine/article/details/123631144)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"] - *2* [对pytorch中的dataset和dataloader的一些理解](https://blog.csdn.net/weixin_45700881/article/details/128351086)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值