706 B
706 B
可以使用 tensorflow.keras.utils.image_dataset_from_directory 來建立 dataset。
dataset 會有 data_batch 與 label_batch 這兩個 member,分別代表資料與標籤。
可以用 dataset.batch(32) 改變 batch size。
還有一些其他的有用function:
shuffle(buffer_size): 打亂順序,可參考Keras.tensorflow - shuffle#^832c8cprefetch(buffer_size): 設定預讀的大小map(callback_func): 用 callback_func 來處理資料take(N): 取出第N筆的批次資料,注意這一筆是一個批次資料,裡面可能有32筆資料(或其他數量,看你的dataset.batch(N)怎麼設定)。
打亂data的方法,請看Keras.tensorflow - shuffle