训练时出错：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 repo（由fizyr指定），但是我再次遇到此错误。

有人可以帮助我吗？谢谢

Answer 1

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