我已经按照pythonprogramming.net的对象检测教程来识别基于ssd_mobilenet_v1_coco模型的小型机器人(我的自定义对象)。
我的机器人大约有450张标记的图像。
我使用ssd_mobilenet_v1_coco的官方示例配置,只进行了必要的更改,如num_class = 1,并将批量大小减少到7,并进行训练,直到我的丢失始终在1到2之间(大约10000个纪元)。
问题是,模型检测到它从预训练状态知道的所有东西都是我的小机器人。因此它将所有东西都识别为机器人,即使它们不是。
我之前遇到过这个问题。并通过添加图像来修复它包含预先训练的对象作为反面例子。另一种解决方法是培训更长时间。如果你这两个都能解决我认为的问题。并尝试增加您的数据集(我正在训练6000张图像)。