Как подготовить набор данных для сверточной нейронной сети в TensorFlow 2.0 (python 3)

Мне понятно как это можно сделать для общедоступных наборов данных, например "mnist"

dataset = tfds.load( 'horses_or_humans' , split=tfds.Split.TRAIN )

Как получить 'dataset' в том же формате для своего набора изображений. Например, в папке "My_Dataset" есть две папки с изображениями, "Class-1" и "Class-2" ?