Как сделать набор данных с изображениями и этикетками?

Question

* Новичок

Я создаю нейронную сеть, которая определяет, является ли автомобиль моделью a или моделью b. Основываясь на этом прогнозе, я хотел бы, чтобы сеть сообщила мне мощность автомобиля. У меня вопрос: как мне создать набор данных с изображением, меткой модели и мощностью. Как мне потом поместить его в массив numpy.

Answer 1

Я предлагаю вам сохранить ваше изображение и его метку (модель a или b) вместе с его функцией (мощностью) в TFRecord, который представляет собой файл, который позволяет вам поддерживать вашу базу данных в готовом к использованию состоянии для Tensorflow.

Именно здесь есть очень хорошее руководство по этой теме: https://www.tensorflow.org/tutorials/load_data/tfrecord

этот ответ на переполнение стека даст вам дополнительную помощь в том, как извлеките данные из файлов TFRecord.