python - 运行 ImageAI 训练总是返回 0 损失

Question

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

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

损失：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

对我来说，解决问题的方法是使用批量大小 > 1。