训练时出错：tensorflow：你的输入数据用完了； 中断训练

Question

我正在尝试执行与以下链接相关联的 colab 笔记本，该链接训练 Keras Retinanet 以查找图像中的对象：

https://www.freecodecamp.org/news/object-detection-in-colab-with-fizyr-retinanet-efed36ac4af3/

但是，即使我完全遵循指南，当我使用以下线路启动火​​车时：

!keras_retinanet/bin/train.py --freeze-backbone --random-transform --weights {PRETRAINED_MODEL} --batch-size 8 --steps 500 --epochs 10 csv annotations.csv classes.csv

我在第一个纪元收到此错误：

WARNING: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, 5000 batches). You may need to use the repeat() function when building your dataset.

即使我重复一遍，我也完全按照笔记本进行操作。

我还尝试使用 Pascal VOC 进行训练，因为在官方 github 存储库（由 fizyr 提供）中指定，但我再次收到此错误。

有人能帮我吗？ 谢谢

Answer 1

编辑：我设法解决了，通过删除 train.py 文件中的默认火车步数，让 keras 自动计算正确的步数。 我在以下线程中找到了hansoli68发布的解决方案： https : //github.com/fizyr/keras-retinanet/issues/1449