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

```pythondset.ImageFolder(root="root folder path", [transform, target_transform])

他有以下成员变量:

  • self.classes - 用一个list保存 类名
  • self.class_to_idx - 类名对应的 索引
  • self.imgs - 保存(img-path, class) tuple的list

results matching ""

    No results matching ""