我已经按照 pythonprogramming.net 的对象检测教程来识别基于 ssd_mobilenet_v1_coco 模型的小型机器人(我的自定义对象)。
我有大约 450 个我的机器人的标记图像。
我使用了 ssd_mobilenet_v1_coco 的官方示例配置,并且只进行了必要的更改,例如 num_class = 1,并将批量大小减少到 7,并训练直到我的损失始终在 1 和 2 之间(大约 10000 个 epoch)。
问题是,该模型从我的小型机器人的预训练状态中检测到它曾经知道的一切。因此,即使它们不是,它也将所有东西都识别为机器人。