![](/img/trans.png)
[英]How to use keras image_dataset_from_directory with custom structures?
[英]How to view the dataset generated by the image_dataset_from_directory function of keras?
import numpy as np
import os
import random
import tensorflow as tf
from tensorflow import keras
import matplotlib.pyplot as plt
%matplotlib inline
CATEGORIES = ['cat', 'dog']
data = keras.preprocessing.image_dataset_from_directory(
'train',
labels = "inferred",
label_mode = "int",
class_names = ['cat', 'dog'],
color_mode = "rgb",
batch_size = 32,
image_size = (256, 256),
shuffle = True,
seed = 42,
validation_split = None,
subset = None,
interpolation = "bilinear",
follow_links = False,
)
我使用上面的 function 來加載數據集並對其進行預處理。 如何查看function返回的數據? print(data)
正在返回創建的 object 的地址,即使將其轉換為 numpy 數組也不起作用。
嘗試:
image_batch, label_batch = next(iter(data))
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.