- 本篇所有代码位置链接????
- 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处理:
结果如下:
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:pytorch 图片处理.md - Python技术站