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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。