如何将图像数据集加载到TensorFlow

Question

我是TensorFlow的新手，我正试图在Google Colaboratory中构建一个小型神经网络。该网络的目标是根据图像识别个人的职业。我有10个不同工作的900张不同的图像作为我的训练数据，还有这些工作的200张不同的图片作为我的测试数据。测试和训练数据都下载到我的电脑上。如有任何帮助，我们将不胜感激。

Answer 1

首先将您的Google Drive安装到Collaborative

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))

有关更多信息，请参阅此链接