Tensorflow对象检测API将旧类检测为我的自定义对象

时间:2018-02-22 12:12:38

标签: tensorflow object-detection object-detection-api

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

我的配置文件,标签地图,测试代码和用于创建tf记录的文件位于:https://github.com/mstferis/tensorflow_object_detection

total loss graphic in tensorboard

使用此模型我将获取这些输出。它会检测手枪(这就是我想要的),但它也会将其他物体检测为手枪。

false example 1

false example 2

false example 3

true example 1

true example 2

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

0 个答案:

没有答案