0

我已经按照 pythonprogramming.net 的对象检测教程来识别基于 ssd_mobilenet_v1_coco 模型的小型机器人(我的自定义对象)。

我有大约 450 个我的机器人的标记图像。

我使用了 ssd_mobilenet_v1_coco 的官方示例配置,并且只进行了必要的更改,例如 num_class = 1,并将批量大小减少到 7,并训练直到我的损失始终在 1 和 2 之间(大约 10000 个 epoch)。

问题是,该模型从我的小型机器人的预训练状态中检测到它曾经知道的一切。因此,即使它们不是,它也将所有东西都识别为机器人。

4

1 回答 1

1

我以前遇到过这个问题。并通过添加包含预训练对象的图像作为反例来修复它。另一种解决方法是训练更长时间。如果你同时做这将解决我认为的问题。并尝试顺便增加您的数据集(我正在使用 6000 张图像进行训练)。

于 2019-02-15T08:56:50.977 回答