代码系列：pytorch——torchvision.datasets.ImageFolder

burning_planet

于 2019-10-16 14:31:40 发布

阅读量1.7k

点赞数 1

分类专栏： Machine Learning python Deep Learning

本文链接：https://blog.youkuaiyun.com/lynlindasy/article/details/102561506

版权

本文介绍了如何使用PyTorch的torchvision.datasets.ImageFolder模块从文件夹结构中加载和处理图像数据。内容包括ImageFolder的属性解析，如classes、class_to_idx、samples和targets，并提供了实例说明，展示了一个包含多个类别的文件夹结构如何被转换为数据集。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

参考博客：https://blog.youkuaiyun.com/TH_NUM/article/details/80877435

用于从文件夹中读取数据，源码链接：https://github.com/pytorch/vision/blob/master/torchvision/datasets/folder.py

ImageFolder是DatasetFolder的子类，有以下属性：

Attributes:

        classes (list): List of the class names.

        class_to_idx (dict): Dict with items (class_name, class_index).

        samples (list): List of (sample path, class_index) tuples

        targets (list): The class_index value for each image in the dataset

self.classes, self.class_to_idx是由方法find_classes得到，self.samples, self.targets由方法make_dataset得到：

classes, class_to_idx = self._find_classes(self.root)
samples = make_dataset(self.root, class_to_idx, extensions, is_valid_file)

最低0.47元/天解锁文章