Python mxnet.gluon.data.vision.datasets.ImageFolderDataset用法及代码示例

用法: class mxnet.gluon.data.vision.datasets.ImageFolderDataset(root, flag=1, transform=None)

root：(str) - 根目录的路径。
flag：({0, 1}, default 1) - 如果为 0，则始终将加载的图像转换为灰度(1 通道)。如果为 1，则始终将加载的图像转换为彩色(3 个通道)。
transform：(callable, default None) -获取数据和标签并转换它们的函数：
```
transform = lambda data, label: (data.astype(np.float32)/255, label)
```

基础：mxnet.gluon.data.dataset.Dataset

用于加载存储在文件夹结构中的图像文件的数据集。

就像:

root/car/0001.jpg
root/car/xxxa.jpg
root/car/yyyb.jpg
root/bus/123.jpg
root/bus/023.jpg
root/bus/wwww.jpg

相关用法

注：本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.gluon.data.vision.datasets.ImageFolderDataset。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。