Tensorflow对象检测API,使用多边形标记的数据集

时间:2017-11-07 03:41:27

标签: tensorflow dataset object-detection-api

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

因此可以修改API,使其可以接受多边形标记的数据集??

1 个答案:

答案 0 :(得分:2)

是的,有可能。您必须提供训练集的目录。但是建议使用边界框,因为在推理期间,您将在检测到的对象周围获得边界框。 You can see an example here in tensorflow.org

对于标签,您可以使用LabelImg,这非常简单易用,也可以提高检测准确度。