TensorFlow对象检测API仅训练一个类而不是多个类

时间:2018-05-03 16:02:24

标签: tensorflow object-detection training-data

我一直在尝试使用我自己的不同饮料数据集来微调SSD Inception v2模型。不幸的是,当我接受培训时,我发现只有一类课程正在接受培训,而其余的则没有。我不确定我做错了什么。我已经重新创建了我的数据集三次并试图用它训练,但我仍然得到相同的结果。

Image that shows the performance by category while training

在训练具有多个类的Object Detection API时,是否有任何特定的过程?

张量流会一个接一个地训练吗?

1 个答案:

答案 0 :(得分:3)

在我的情况下,问题是由配置文件引起的。我已经为num_classes分配了一个而不是三个。在我用三个初始化之后,一切都运行良好。有时,由于数据集的问题,也可能出现这种问题。