➜  torch_test tree animal_data
animal_data
├── train
│   ├── ants
│   │   ├── 0013035.jpg
│   │   ├── 1030023514_aad5c608f9.jpg
│   │   ├── 1095476100_3906d8afde.jpg
│   └── bees
│       ├── 1092977343_cb42b38d62.jpg
│       ├── 1093831624_fb5fbe2308.jpg
│       ├── 1097045929_1753d1c765.jpg
└── val
    ├── ants
    │   ├── 10308379_1b6c72e180.jpg
    │   ├── 1053149811_f62a3410d3.jpg
    │   ├── 1073564163_225a64f170.jpg
    └── bees
        ├── 1032546534_06907fe3b3.jpg
        ├── 759745145_e8bc776ec8.jpg
        └── abeja.jpg
  • 加载图片处理,用到 datasets.ImageFolder() 方法,接受的参数有(path, transform, target_transform),该方法源代码如下:
    pytorch 图片处理.md

  • 加载方法如下:
    pytorch 图片处理.md

  • transform处理后的结果如下:
    pytorch 图片处理.md

  • 对transform后的数据做进一步处理,DataLoader处理:
    pytorch 图片处理.md
    结果如下:
    pytorch 图片处理.md