當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文


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