Tensorflow 2.0 小图像数据集警告

Question

我正在使用 TensorFlow 2.0 学习图像分类，在实施它时收到此警告：

tensorflow:Your input ran out of data; interrupting training. Make sure that your dataset or generator can generate at least steps_per_epoch * epochs batches (in this case, 11280 batches). You may need to use the repeat() function when building your dataset.

以下是数据集大小和模型数据：

total training mango images : 752
total validation mango images : 288

history = model.fit_generator(train_generator,
                              validation_data=validation_generator,
                              steps_per_epoch=752,
                              epochs=15,
                              validation_steps=288)

我还使用了图像增强技术来提高模型的效率。但是，警告仍然会中断训练并且模型仍然过拟合。

Here is the link to my full model in github

请帮帮我！

提前谢谢你

Answer 1

鉴于您在第 24 步出现错误并且 batch_size 是 32，我想问题是您的第 24 批只有 16 张图像。您可以删除这些图像，以便您的数据集与批处理大小 32 对齐，或将 batch_size 更改为 16。

顺便说一句，我怀疑你使用的 steps_per_epoch 和 validation_steps 是错误的 - "steps" 表示批次数，而不是样本数。仅供参考

Tensorflow 2.0 小图像数据集警告

Tensorflow 2.0 small image dataset warning

image

tensorflow-datasets

tensorflow2.0