【发布时间】:2021-11-13 16:37:31
【问题描述】:
ImageFolder 数据集有标签吗? ImageFolder 数据集中存储了哪些详细信息?
当我尝试获取 ImageFolderdataset 的标签时出现此错误
AttributeError: 'Subset' object has no attribute 'labels'
【问题讨论】:
标签: python image-processing pytorch
ImageFolder 数据集有标签吗? ImageFolder 数据集中存储了哪些详细信息?
当我尝试获取 ImageFolderdataset 的标签时出现此错误
AttributeError: 'Subset' object has no attribute 'labels'
【问题讨论】:
标签: python image-processing pytorch
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.png、dog/xxy.png 以及/dog 下的所有其他图像都将被分配标签0(此处对应于"dog" 类)。
同样,cat/123.png、cat/nsdf3.png 以及/cat 下的所有其他图像将被分配不同的标签,此处为1。
【讨论】:
(image tensor, label integer) tuple 对。