Pytorch中Dataloader保存文件名

转载自：https://gist.github.com/andrewjong/6b02ff237533b3b2c554701fb53d5c4d，本文只做个人记录学习使用，版权归原作者所有。

import torch
from torchvision import datasetsclass ImageFolderWithPaths(datasets.ImageFolder):"""Custom dataset that includes image file paths. Extendstorchvision.datasets.ImageFolder"""# override the __getitem__ method. this is the method that dataloader callsdef __getitem__(self, index):# this is what ImageFolder normally returns original_tuple = super(ImageFolderWithPaths, self).__getitem__(index)# the image file pathpath = self.imgs[index][0]# make a new tuple that includes original and the pathtuple_with_path = (original_tuple + (path,))return tuple_with_path# EXAMPLE USAGE:
# instantiate the dataset and dataloader
data_dir = "your/data_dir/here"
dataset = ImageFolderWithPaths(data_dir) # our custom dataset
dataloader = torch.utils.DataLoader(dataset)# iterate over data
for inputs, labels, paths in dataloader:# use the above variables freelyprint(inputs, labels, paths)

Pytorch中Dataloader保存文件名相关推荐

Pytorch中Dataloader踩坑：RuntimeError: DataLoader worker (pid(s) 6700, 10620) exited unexpectedly
Pytorch中Dataloader踩坑环境: 问题背景: 观察报错信息进行分析根据分析进行修改尝试总结环境: 系统:windows10 Pytorch版本:1.5.1+cu101 问题背景: ...
pytorch中DataLoader的num_workers参数详解与设置大小建议
Q:在给Dataloader设置worker数量(num_worker)时,到底设置多少合适?这个worker到底怎么工作的? train_loader = torch.utils.data.Data ...
pytorch中Dataloader()中的num_workers设置问题
pytorch中Dataloader()中的num_workers设置问题: 如果num_workers的值大于0,要在运行的部分放进__main__()函数里,才不会有错: import numpy ...
Pytorch中DataLoader类
1 基本流程首先熟知,pytorch 的数据加载到模型的操作顺序是这样的: 创建一个 Dataset 对象创建一个 DataLoader 对象循环这个 DataLoader 对象,将img, l ...
pytorch中dataloader的num_workers参数
结论速递在Windows系统中,num_workers参数建议设为0,在Linux系统则不需担心. 1 问题描述在之前的任务超大图上的节点表征学习中,使用PyG库用于数据加载的DataLoader ...
pytorch中DataLoader的num_workers
Question 一直很迷, 在给Dataloader设置worker数量(num_worker)时,到底设置多少合适?这个worker到底怎么工作的? 如果将num_worker设为0(也是默认值) ...
pytorch中调整学习率的lr_scheduler机制
pytorch中调整学习率的lr_scheduler机制 </h1><div class="clear"></div><div class ...
【小白学习PyTorch教程】五、在 PyTorch 中使用 Datasets 和 DataLoader 自定义数据
「@Author:Runsen」有时候,在处理大数据集时,一次将整个数据加载到内存中变得非常难. 因此,唯一的方法是将数据分批加载到内存中进行处理,这需要编写额外的代码来执行此操作.对此,PyTor ...
Pytorch 中的数据类型 torch.utils.data.DataLoader 参数详解
DataLoader是PyTorch中的一种数据类型,它定义了如何读取数据方式.详情也可参考本博主的另一篇关于torch.utils.data.DataLoader(https://blog.csdn ...

Pytorch中Dataloader保存文件名

Pytorch中Dataloader保存文件名相关推荐

最新文章

热门文章