2

我是机器学习的初学者,我正在尝试使用自己的数据集进行自己的对象检测。但是,如果对象用多边形边界标记会更实用。然而 tensorflow 对象检测 API 只能接受边界框。

那么是否可以修改 API 使其可以接受多边形标记的数据集?

4

1 回答 1

2

对的,这是可能的。您必须提供训练集的目录。但推荐使用边界框,因为在推理期间,您会在检测到的对象周围获得边界框。您可以在 tensorflow.org 中看到一个示例

对于标签,您可以使用LabelImg,它非常简单易用,也将提高检测精度。

于 2017-11-07T05:03:58.957 回答