4

我正在尝试使用 Tensorflow 对象检测 API 创建自定义对象检测器。我遵循了 API 文档和许多其他教程。我正在使用预先训练的更快的 rcnn 模型和默认配置文件值(我正确编辑了“要配置的路径”区域)(我的标签映射只有一个项目,它的 id 是 1)。在使用 6660 个训练示例进行训练后,结果并不完美,但对于我的测试图像来说还可以。主要问题是当我使用不包含我的自定义对象的不同图像测试模型时,它会将其他对象检测为我的自定义对象。特别是,如果对象在 coco 数据集(我的模型在 coco 数据集上预训练)中,比如汽车、刀、笔记本电脑等。模型检测它们的位置并将它们标记为我的自定义对象。这是一个错误还是我错过了什么?

我用于创建 tf 记录的配置文件、标签映射、测试代码和文件在这里:https ://github.com/mstferis/tensorflow_object_detection

张量板中的总损失图形

有了这个模型,我得到了这些输出。它检测到手枪(这就是我想要的),但它也将其他物体检测为手枪。

错误示例 1

错误示例 2

错误示例 3

真实例子1

真实例子2

*注意(更新):最后我将其他类的图片添加到我的训练集中,并将它们标记为我的自定义对象。现在模型预测的误报更少。但是我仍然找不到以前的问题。

4

0 回答 0