将图像数据集加载到 pytorch 中的数据加载器后，如何将图像数据集下载到我的电脑？

【问题标题】：How can I download to my pc an image dataset after loading it into a dataloader in pytorch?将图像数据集加载到 pytorch 中的数据加载器后，如何将图像数据集下载到我的电脑？
【发布时间】：2021-11-30 13:08:50
【问题描述】：

我一直在关注这篇文章： https://github.com/moein-shariatnia/Deep-Learning/tree/main/Image%20Colorization%20Tutorial

我有一个问题。我正在尝试下载已加载到 pytorch 数据加载器中的图像，但由于它们是从更大的数据集中随机选择的，我无法弄清楚。在选择它们后，如何将已加载到“train_dl”中的确切图像下载到我的 PC（作为文件夹）？我已经在这些精确的图像上训练了我的模型，所以我将需要它们用于未来的工作。

【问题讨论】：

标签： machine-learning pytorch pytorch-dataloader

【解决方案1】：

Dataloaders 应该有一个 dataset 属性，您可以使用它来查找代码中的路径

在代码中：

class ColorizationDataset(Dataset):
    def __init__(self, paths, split='train'):
        ...
        
        self.split = split
        self.size = SIZE
        self.paths = paths #<<<<- IMPORTANT LINE HERE
    
    def __getitem__(self, idx):
        ...
    
    def __len__(self):
        return len(self.paths)

def make_dataloaders(batch_size=16, n_workers=4, pin_memory=True, **kwargs): # A handy function to make our dataloaders
    dataset = ColorizationDataset(**kwargs)
    dataloader = DataLoader(dataset, batch_size=batch_size, num_workers=n_workers,
                            pin_memory=pin_memory)
    return dataloader

因此，要从数据加载器中获取项目，您可以运行：

train_paths = train_dl.dataset.paths

您可以循环浏览此列表以单独保存图像（不推荐）或将其保存为文本，将训练/验证/测试拆分为单独的文件。

【讨论】：