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

Question

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

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

Answer 1

是的，这是可能的。你必须给出训练集的目录。但建议使用边界框，因为在推理期间，您会在检测到的对象周围得到边界框。 You can see an example here in tensorflow.org.

标注可以使用LabelImg，非常简单易用，也能提高检测准确率。