pytorch加载自己的数据集,重写Dataset类

len 方法,提供了dataset的大小; getitem 方法, 该方法支持从 0len(self)的索引。

 #filenames是训练数据文件名称列表,labels是标签列表
class MyDataset(Dataset):
    def __init__(self, filenames, labels, transform):
        self.filenames = filenames
        self.labels 
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
PyTorch,有几种方法可以加载自己的数据集。一种常见的方法是使用`torch.utils.data.Dataset`创建自定义数据集。你可以创建一个,继承自`torch.utils.data.Dataset`,并重写`__len__()`和`__getitem__()`方法来定义数据集的大小和获取样本的方式。例如,你可以参考以下代码示例: ```python import torch from torch.utils.data import Dataset class MyDataset(Dataset): def __init__(self, data, targets): self.data = data self.targets = targets def __len__(self): return len(self.data) def __getitem__(self, index): x = self.data[index] y = self.targets[index] return x, y ``` 另一种方法是使用`torch.utils.data.DataLoader`加载数据集。`DataLoader`可以将自定义数据集(如上面的`MyDataset`)包装成一个可迭代的数据加载器。你可以指定批量大小、是否打乱数据等参数来定制数据加载器。通过数据加载器,你可以轻松地遍历整个数据集并获取批量的样本。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* *3* [PyTorch加载自己的数据集](https://blog.csdn.net/fb_941219/article/details/129495181)[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_2"}}] [.reference_item style="max-width: 50%"] - *2* [PyTorch如何加载数据集自定义数据集)](https://blog.csdn.net/a1819797401/article/details/98872840)[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_2"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

绛洞花主敏明

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值