тензор потока использовать путь изображения в качестве учебного ввода - PullRequest
0 голосов
/ 29 марта 2020

Можно ли использовать список, содержащий относительный путь ко всем файлам тренировочных образов, в качестве входных данных при запуске model.fit?

Например, если массив имеет вид:

['images/00001.png', 'images/00002.png', ... 'images/9999.png']

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

1 Ответ

0 голосов
/ 29 марта 2020

Попробуйте сохранить ваши изображения в кортеже. Он хорошо работает с Tensorflow и может хранить много данных при необходимости.

thistuple = ("images/00001.png", "images/00002.png", "images/9999.png")
print(thistuple)

РЕДАКТИРОВАТЬ

Создать ImageDataGenerator в Keras

#create a data generator
datagen = ImageDataGenerator()

Затем используйте этот генератор для загрузки изображений

#load and iterate training dataset
train_it = datagen.flow_from_directory('data/train/', class_mode='binary', 
batch_size=64)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...