Подготовка изображений для подачи в тензорный поток в виде наборов данных

Question

AskTensorFlow

Я использовал наборы данных с тензорным потоком, такие как MNIST, IMDB, для изучения работы тензорного потока. Однако в практических приложениях мы должны предварительно обработать и подготовить набор данных самостоятельно. Предположим, я работаю с набором данных изображений, поэтому я хочу предварительно обработать их до формата, который может быть передан в модель тензорного потока. Как я могу предварительно обработать набор данных изображения в тензорном формате?

Answer 1

При работе с изображениями вы обычно будете использовать генератор.
Генератор - это функция, которая выводит (u,v), где u - это сэмплы, а v - это метки.

Пример того, как это сделать, можно найти здесь Как обучить сеть TensorFlow с использованием генератора для получения входных данных? .

При создании функции генератора для работы с изображением помните, что каждое изображение является просто массивом: либо (x,y) для оттенков серого, либо (x,y,channels) для цветного изображения.
Таким образом, функция вашего генератора должна будет прочитать пакет изображений с диска и превратить их в массивы.Для этого есть множество инструментов: opencv , scipy , PIL .
После загрузки изображений вы можете делать с ними любые манипуляции (с помощью этих или других инструментов) обычно вам нужно изменить форму изображения в соответствии с вашей моделью.
В итоге вам нужно будет вывести пару ([batch_size,x,y,channels], [batch_size,labels]).