pytorch 图片处理.md

• 本篇所有代码位置链接🔗
• pytorch 图片处理，主要用到 torchvision 模块的 datasets 和 transforms。
• 例如：本地图片资源目录结构如下

➜  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），该方法源代码如下：
  

• 加载方法如下：
  

• transform处理后的结果如下：
  

• 对transform后的数据做进一步处理，DataLoader处理：
  
  结果如下：