keras-team / autokeras

AutoML library for deep learning

Home Page:http://autokeras.com/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Feature: image_dataset_from_directory support one_hot/categorical labels

skyetomez opened this issue · comments

tensorflow keras has this built in but it not available currently.

Code Example

ak.keras.utils.image_dataset_from_directory(
    directory,
    labels='inferred',
    label_mode='int',
    class_names=None,
    color_mode='rgb',
    batch_size=32,
    image_size=(256, 256),
    shuffle=True,
    seed=None,
    validation_split=None,
    subset=None,
    interpolation='bilinear',
    follow_links=False,
    crop_to_aspect_ratio=False,
    **kwargs
)

Reason

This would make using he ak generator of image datasets from a directory more convenient