0

我正在尝试执行与以下链接相关联的 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 指定）中指定，但我再次收到此错误。

有人能帮我吗？谢谢

4

1 回答 1

0

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

于 2020-10-12T20:47:40.463 回答