tensorflow - 值错误 : Expect x to be a non-empty array or dataset (Tensor Flow lite model maker on Collab)

Question

我正在关注 this在 Collab 上使用 TensorFlow lite Model Maker 创建自定义模型的教程。

import pathlib
path = pathlib.Path('/content/employee_pics') 
count = len(list(path.glob('*/*.jpg')))
count

data = ImageClassifierDataLoader.from_folder(path)
train_data, test_data = data.split(0.5)

我有第 2 步的问题:

model = image_classifier.create(train_data)

我收到一个错误:
值错误: 期望 x 是一个非空数组或数据集。

难道我做错了什么？不过，示例中提供的数据集运行良好。为什么？

Answer 1

此错误是由训练数据的大小小于 batch_size 引起的这是不允许的。
默认batch_size为32，即训练图片数量不少于32张。不需要计算每个标签的图片数量，只需要保证训练图片总数至少为32张即可。
您需要选择以下解决方案之一来解决它。

套装batch_size小于训练数据大小，例如:

image_classifier.create(train_data, batch_size=4)

通过添加更多数据来增加训练数据的大小。