訓練時出錯：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