【问题标题】:ImageFolder datasetImageFolder 数据集
【发布时间】:2021-11-13 16:37:31
【问题描述】:

ImageFolder 数据集有标签吗? ImageFolder 数据集中存储了哪些详细信息?

当我尝试获取 ImageFolderdataset 的标签时出现此错误

AttributeError: 'Subset' object has no attribute 'labels'

【问题讨论】:

    标签: python image-processing pytorch


    【解决方案1】:

    ImageFolder 专为分类任务而设计,您的目录结构决定了为每个图像分配的标签。

    文档页面上给出的示例如下:

    root/dog/xxx.png
    root/dog/xxy.png
    root/dog/[...]/xxz.png
    
    root/cat/123.png
    root/cat/nsdf3.png
    root/cat/[...]/asd932_.png
    

    图像dog/xxx.pngdog/xxy.png 以及/dog 下的所有其他图像都将被分配标签0(此处对应于"dog" 类)。

    同样,cat/123.pngcat/nsdf3.png 以及/cat 下的所有其他图像将被分配不同的标签,此处为1

    【讨论】:

    • 感谢您的简短说明。有什么方法可以检索 0,1 值。​​
    • 如果您初始化数据集并对其进行迭代,它将产生(image tensor, label integer) tuple 对。
    猜你喜欢
    • 2020-09-28
    • 1970-01-01
    • 2020-04-04
    • 2022-08-26
    • 2019-06-09
    • 1970-01-01
    • 2020-04-27
    • 1970-01-01
    • 2021-09-03
    相关资源
    最近更新 更多