运行ImageAI训练始终返回0损失

Question

我正在尝试使用典型的全息教程来使用Tensorflow、Keras和ImageAI来训练新的自定义对象检测。

训练正在运行，但它总是返回如下内容：

损失: 0.0000e+00 - yolo_layer_1_loss: 0.0000e+00 - yolo_layer_2_loss: 0.0000e+00

当我尝试使用模型的结果时，它失败了。

我使用的是Python 3.7.7 tensorflow 1.14.0

imageai 2.1.5

Keras 2.3.1

Answer 1

对我来说，解决这个问题的方法是使用batch size > 1。