Пользовательский набор данных в TensorFlow

Question

У меня есть несколько изображений и соответствующие метки в качестве обучающего набора.

Я хочу использовать его в качестве набора данных в TensorFlow (как то, как мы используем набор данных mnist и можем получать данные в пакетном режиме с помощью mnist.train.next_batch (BATCH_SIZE))

Вопрос в том, как мне создать свой собственный набор данных, чтобы я мог использовать его в пакетах?

Дополнительная информация: в тренировочном наборе 4575 изображений по 28 x28 (в масштабе) и классификационные метки. Всего 62 класса.

Пожалуйста, дайте мне знать, если требуется дополнительная информация.

Answer 1

Вы можете использовать tf.data.from_tensor_slices(tf.convert_to_tensor(np_images), tf.convert_to_tensor(np_labels)) для создания набора данных, состоящего из пар (изображение, метка). Теперь вы можете применить к этому .batch(BATCHSIZE) функцию, которая будет создавать пакеты нужного вам размера. Затем вы можете просто напрямую передать это функции модели, например .fit(), если используете tf.keras.

. Для получения дополнительной информации перейдите по следующим ссылкам -

from_tensor_slices

convert_to_tensor

партия