该模型将其预先训练过的模型所知道的一切都误认为是我的自定义对象

Question

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

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

我使用ssd_mobilenet_v1_coco的官方示例配置，只进行了必要的更改，如num_class = 1，并将批量大小减少到7，并进行训练，直到我的丢失始终在1到2之间（大约10000个纪元）。

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

Answer 1

我之前遇到过这个问题。并通过添加图像来修复它包含预先训练的对象作为反面例子。另一种解决方法是培训更长时间。如果你这两个都能解决我认为的问题。并尝试增加您的数据集（我正在训练6000张图像）。