FBAGSTM's picture
STM32 AI Experimentation Hub
747451d
# /*---------------------------------------------------------------------------------------------
# * Copyright (c) 2022 STMicroelectronics.
# * All rights reserved.
# * This software is licensed under terms that can be found in the LICENSE file in
# * the root directory of this software component.
# * If no LICENSE file comes with this software, it is provided AS-IS.
# *--------------------------------------------------------------------------------------------*/
from .utils import load_cifar_batch, get_train_val_ds, get_ds, get_prediction_ds, \
preprocess_data, prepare_kwargs_for_dataloader
from .cifar10 import load_cifar10
from .cifar100 import load_cifar100
from .emnist_byclass import load_emnist
from .imagenet import load_imagenet_like
from .custom_dataset import load_custom_dataset