Как загрузить наборы данных изображений в TensorFlow?

Question

Я новичок в TensorFlow и пытаюсь построить небольшую нейронную сеть в Google Colab Laboratory. Цель этой сети состоит в том, чтобы идентифицировать занятие человека на основе изображения. У меня есть 900 различных изображений для каждого из 10 различных заданий в качестве данных обучения, и 200 различных изображений для каждого из этих заданий в качестве данных тестирования. Данные тестирования и обучения загружаются на мой компьютер. Любая помощь будет принята с благодарностью.

Answer 1

Сначала подключите Google Drive к Collaboratory

from google.colab import drive
drive.mount('/content/drive')

Путь к каталогам, в которых присутствуют данные обучения и тестирования

train_dir  = '/content/drive/Training'
test_dir = '/content/drive/Testing'

Создание генератора данных для обучения и тестирования

train_datagen = ImageDataGenerator(**datagen_kwargs)
test_datagen = ImageDataGenerator(**datagen_kwargs)

train_data = train_datagen.flow_from_directory(
    train_dir,
    target_size = (img_width, img_height),
    shuffle=True,
    batch_size = batch_size,
    classes = list(class_names))

test_data = test_datagen.flow_from_directory(
    test_dir,
    target_size = (img_width, img_height),
    shuffle=True,
    batch_size = batch_size,
    classes = list(class_names))

Для получения дополнительной информации, пожалуйста, перейдите по этой ссылке