Tensorflow对象检测API如何添加背景类示例?

时间:2017-11-20 11:27:45

标签: tensorflow object-detection

我正在使用tensorflow对象检测API。我有两类兴趣。在第一次试验中,我得到了合理的结果,但我发现在纯背景图像中很容易得到两个类的误报。这些背景图像(即没有任何类bbx的图像)尚未包括在训练集中。

如何将它们添加到训练集中?如果我只是添加没有bbx的样本,似乎无效。

1 个答案:

答案 0 :(得分:0)

您的目标是将负图像添加到您的训练数据集中以加强背景类(检测 API 中的 id 0)。您可以使用 VOC Pascal XML 注释格式来实现这一点。在您的 XML 文件中是没有对象的图像的高度和宽度。通常,您标记对象的坐标、高度和宽度以及对象名称都在 XML 文件中。如果您使用 labelImg,您可以使用验证按钮生成与您的负片图像相对应的 XML 文件。 Roboflow 也可以生成带对象和不带对象的 XML 文件。